From 9483db15b567dd56dc081465817548df11f4bac1 Mon Sep 17 00:00:00 2001 From: Lester Hedges Date: Tue, 10 Dec 2024 11:08:12 +0000 Subject: [PATCH] Blacken. [ci skip] --- emle/models/_mace.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/emle/models/_mace.py b/emle/models/_mace.py index cb50662..b05cde2 100644 --- 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