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/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)",