lint

mosaicml · Jan 9, 2024 · 50fd5e2 · 50fd5e2
1 parent 04279ca
commit 50fd5e2
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/algorithms/test_required_on_load.py b/tests/algorithms/test_required_on_load.py
@@ -165,7 +165,9 @@ def test_autoload(algo_name: str, load_weights_only: bool, already_added: bool,
         # Excluding some algorithms leads to errors when loading
         elif exclude:
             if version.parse(torch.__version__) > version.parse('2.1.3'):
-                if algo_name in ['Alibi', 'BlurPool', 'Factorize', 'GatedLinearUnits', 'GhostBatchNorm', 'SqueezeExcite']:
+                if algo_name in [
+                        'Alibi', 'BlurPool', 'Factorize', 'GatedLinearUnits', 'GhostBatchNorm', 'SqueezeExcite'
+                ]:
                     context = pytest.raises(KeyError)  # Optimizer loading is strict
             else:
                 if algo_name in ['Factorize', 'SqueezeExcite']: