format

matsengrp · Nov 13, 2024 · 28d7249 · 28d7249
1 parent 268b713
commit 28d7249
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/test_pretrained.py b/tests/test_pretrained.py
@@ -5,7 +5,9 @@
 
 def test_names_unique():
     # Check all defined models can be loaded and they have unique names
-    assert len(set(pretrained.MODEL_TO_LOCAL.keys())) == sum(len(models) for _, _, _, models in pretrained.PACKAGE_LOCATIONS_AND_CONTENTS)
+    assert len(set(pretrained.MODEL_TO_LOCAL.keys())) == sum(
+        len(models) for _, _, _, models in pretrained.PACKAGE_LOCATIONS_AND_CONTENTS
+    )
 
 
 def test_load_all_models():