diff --git a/examples/movement-pruning/bertarize.py b/examples/movement-pruning/bertarize.py index 7b6b5b83..d1e2462a 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 183c125c..8553f6f8 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 6646667e..09c90016 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 761a6343..c1bfd139 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