Commit 7dce8dc7 authored by Aymeric Augustin's avatar Aymeric Augustin
Browse files

Fix E731 flake8 warning (x3).

parent eed46f38
...@@ -184,7 +184,10 @@ def save_rouge_scores(str_scores): ...@@ -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: ...@@ -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): ...@@ -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
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment