Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
chenpangpang
transformers
Commits
7dce8dc7
Commit
7dce8dc7
authored
Dec 21, 2019
by
Aymeric Augustin
Browse files
Fix E731 flake8 warning (x3).
parent
eed46f38
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
8 additions
and
3 deletions
+8
-3
examples/summarization/run_summarization.py
examples/summarization/run_summarization.py
+4
-1
transformers/commands/serving.py
transformers/commands/serving.py
+4
-1
transformers/modeling_utils.py
transformers/modeling_utils.py
+0
-1
No files found.
examples/summarization/run_summarization.py
View file @
7dce8dc7
...
@@ -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
...
...
transformers/commands/serving.py
View file @
7dce8dc7
...
@@ -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
...
...
transformers/modeling_utils.py
View file @
7dce8dc7
...
@@ -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
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment