flake8 compliance

2020-05-28 00:37:09 -04:00
parent b11386e158
commit cc746a5020
4 changed files with 4 additions and 2 deletions
--- 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)",