Blacken. [ci skip]

chemle · Dec 10, 2024 · 9483db1 · 9483db1
1 parent b16801f
commit 9483db1
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/emle/models/_mace.py b/emle/models/_mace.py
@@ -199,7 +199,9 @@ def __init__(
                         f"Unsupported MACE model: '{mace_model}'. Available MACE-OFF23 models are "
                         "'mace-off23-small', 'mace-off23-medium', 'mace-off23-large'"
                     )
-                source_model = _mace_off(model=size, device=device, return_raw_model=True)
+                source_model = _mace_off(
+                    model=size, device=device, return_raw_model=True
+                )
             else:
                 # Assuming that the model is a local model.
                 if _os.path.exists(mace_model):
@@ -208,7 +210,9 @@ def __init__(
                     raise FileNotFoundError(f"MACE model file not found: {mace_model}")
         else:
             # If no MACE model is provided, use the default MACE-OFF23(S) model.
-            source_model = _mace_off(model="small", device=device, return_raw_model=True)
+            source_model = _mace_off(
+                model="small", device=device, return_raw_model=True
+            )
 
         from mace.tools.scripts_utils import extract_config_mace_model