flake8 compliance

This commit is contained in:
Victor SANH committed 2020-06-01 09:23:31 -04:00
1 parent b11386e158
commit cc746a5020
4 files changed
+4 -2

No files matched your search

+1 -1
View File
@@ -100,7 +100,7 @@ if __name__ == "__main__":
parser.add_argument( parser.add_argument(
"--pruning_method", "--pruning_method",
choices=["l0", "magnitude", "topK", "sigmoied_threshold",], choices=["l0", "magnitude", "topK", "sigmoied_threshold"],
type=str, type=str,
required=True, required=True,
help="Pruning Method (l0 = L0 regularization, magnitude = Magnitude pruning, topK = Movement pruning, sigmoied_threshold = Soft movement pruning)", help="Pruning Method (l0 = L0 regularization, magnitude = Magnitude pruning, topK = Movement pruning, sigmoied_threshold = Soft movement pruning)",
@@ -67,7 +67,7 @@ if __name__ == "__main__":
parser.add_argument( parser.add_argument(
"--pruning_method", "--pruning_method",
choices=["l0", "topK", "sigmoied_threshold",], choices=["l0", "topK", "sigmoied_threshold"],
type=str, type=str,
required=True, required=True,
help="Pruning Method (l0 = L0 regularization, topK = Movement pruning, sigmoied_threshold = Soft movement pruning)", help="Pruning Method (l0 = L0 regularization, topK = Movement pruning, sigmoied_threshold = Soft movement pruning)",
@@ -1,3 +1,4 @@
# flake8: noqa
from .configuration_bert_masked import MaskedBertConfig from .configuration_bert_masked import MaskedBertConfig
from .modeling_bert_masked import ( from .modeling_bert_masked import (
MaskedBertForMultipleChoice, MaskedBertForMultipleChoice,
@@ -1,2 +1,3 @@
# flake8: noqa
from .binarizer import MagnitudeBinarizer, ThresholdBinarizer, TopKBinarizer from .binarizer import MagnitudeBinarizer, ThresholdBinarizer, TopKBinarizer
from .masked_nn import MaskedLinear from .masked_nn import MaskedLinear