Skip to content

Commit 11f324f

Browse files
committed
feat: update model interface
1 parent 8761d44 commit 11f324f

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

lambench/models/ase_models.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -82,7 +82,7 @@ def __init__(self, *args, **kwargs):
8282
self._calc = None
8383

8484
@property
85-
def calc(self, head=None) -> Calculator:
85+
def calc(self) -> Calculator:
8686
"""ASE Calculator with the model loaded."""
8787
calculator_dispatch = {
8888
"MACE": self._init_mace_calculator,

lambench/models/models_config.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -300,4 +300,4 @@
300300
# packages:
301301
# pet-mad: 1.0.0
302302
# show_direct_task: False # Hide deprecated model
303-
# skip_tasks: [CalculatorTask, DirectPredictTask]
303+
# skip_tasks: [CalculatorTask, DirectPredictTask]

0 commit comments

Comments
 (0)