From cc746a502032434279702a5ffd2a41443c8def48 Mon Sep 17 00:00:00 2001 From: Victor SANH Date: Thu, 28 May 2020 00:37:09 -0400 Subject: [PATCH] flake8 compliance --- examples/movement-pruning/bertarize.py | 2 +- examples/movement-pruning/counts_parameters.py | 2 +- examples/movement-pruning/emmental/__init__.py | 1 + examples/movement-pruning/emmental/modules/__init__.py | 1 + 4 files changed, 4 insertions(+), 2 deletions(-) diff --git a/examples/movement-pruning/bertarize.py b/examples/movement-pruning/bertarize.py index 7b6b5b834b..d1e2462a30 100644 --- a/examples/movement-pruning/bertarize.py +++ b/examples/movement-pruning/bertarize.py @@ -100,7 +100,7 @@ if __name__ == "__main__": parser.add_argument( "--pruning_method", - choices=["l0", "magnitude", "topK", "sigmoied_threshold",], + choices=["l0", "magnitude", "topK", "sigmoied_threshold"], type=str, required=True, help="Pruning Method (l0 = L0 regularization, magnitude = Magnitude pruning, topK = Movement pruning, sigmoied_threshold = Soft movement pruning)", diff --git a/examples/movement-pruning/counts_parameters.py b/examples/movement-pruning/counts_parameters.py index 183c125c4a..8553f6f812 100644 --- a/examples/movement-pruning/counts_parameters.py +++ b/examples/movement-pruning/counts_parameters.py @@ -67,7 +67,7 @@ if __name__ == "__main__": parser.add_argument( "--pruning_method", - choices=["l0", "topK", "sigmoied_threshold",], + choices=["l0", "topK", "sigmoied_threshold"], type=str, required=True, help="Pruning Method (l0 = L0 regularization, topK = Movement pruning, sigmoied_threshold = Soft movement pruning)", diff --git a/examples/movement-pruning/emmental/__init__.py b/examples/movement-pruning/emmental/__init__.py index 6646667ea8..09c900161d 100644 --- a/examples/movement-pruning/emmental/__init__.py +++ b/examples/movement-pruning/emmental/__init__.py @@ -1,3 +1,4 @@ +# flake8: noqa from .configuration_bert_masked import MaskedBertConfig from .modeling_bert_masked import ( MaskedBertForMultipleChoice, diff --git a/examples/movement-pruning/emmental/modules/__init__.py b/examples/movement-pruning/emmental/modules/__init__.py index 761a6343d6..c1bfd1397c 100644 --- a/examples/movement-pruning/emmental/modules/__init__.py +++ b/examples/movement-pruning/emmental/modules/__init__.py @@ -1,2 +1,3 @@ +# flake8: noqa from .binarizer import MagnitudeBinarizer, ThresholdBinarizer, TopKBinarizer from .masked_nn import MaskedLinear