Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
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