Fix E731 flake8 warning (x3).
This commit is contained in:
@@ -184,7 +184,10 @@ def save_rouge_scores(str_scores):
|
|||||||
def build_data_iterator(args, tokenizer):
|
def build_data_iterator(args, tokenizer):
|
||||||
dataset = load_and_cache_examples(args, tokenizer)
|
dataset = load_and_cache_examples(args, tokenizer)
|
||||||
sampler = SequentialSampler(dataset)
|
sampler = SequentialSampler(dataset)
|
||||||
collate_fn = lambda data: collate(data, tokenizer, block_size=512, device=args.device)
|
|
||||||
|
def collate_fn(data):
|
||||||
|
return collate(data, tokenizer, block_size=512, device=args.device)
|
||||||
|
|
||||||
iterator = DataLoader(dataset, sampler=sampler, batch_size=args.batch_size, collate_fn=collate_fn,)
|
iterator = DataLoader(dataset, sampler=sampler, batch_size=args.batch_size, collate_fn=collate_fn,)
|
||||||
|
|
||||||
return iterator
|
return iterator
|
||||||
|
|||||||
@@ -15,7 +15,10 @@ try:
|
|||||||
_serve_dependancies_installed = True
|
_serve_dependancies_installed = True
|
||||||
except (ImportError, AttributeError):
|
except (ImportError, AttributeError):
|
||||||
BaseModel = object
|
BaseModel = object
|
||||||
Body = lambda *x, **y: None
|
|
||||||
|
def Body(*x, **y):
|
||||||
|
pass
|
||||||
|
|
||||||
_serve_dependancies_installed = False
|
_serve_dependancies_installed = False
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -77,7 +77,6 @@ class PreTrainedModel(nn.Module):
|
|||||||
"""
|
"""
|
||||||
config_class = None
|
config_class = None
|
||||||
pretrained_model_archive_map = {}
|
pretrained_model_archive_map = {}
|
||||||
load_tf_weights = lambda model, config, path: None
|
|
||||||
base_model_prefix = ""
|
base_model_prefix = ""
|
||||||
|
|
||||||
@property
|
@property
|
||||||
|
|||||||
Reference in New Issue
Block a user