Skip to content

Commit f2261ef

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 7860c66 commit f2261ef

4 files changed

Lines changed: 6 additions & 4 deletions

File tree

lambench/metrics/results/results.json

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3513,4 +3513,4 @@
35133513
]
35143514
}
35153515
}
3516-
}
3516+
}

lambench/metrics/results/scatter.json

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -95,4 +95,4 @@
9595
"std": 135.56,
9696
"generalizability error": 0.34
9797
}
98-
]
98+
]

lambench/models/ase_models.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -103,7 +103,9 @@ def _init_orb_calculator(self) -> Calculator:
103103
from orb_models.forcefield import pretrained
104104
from orb_models.forcefield.calculator import ORBCalculator
105105

106-
orbff = pretrained.ORB_PRETRAINED_MODELS[self.model_name.replace('_', '-')](device="cuda")
106+
orbff = pretrained.ORB_PRETRAINED_MODELS[self.model_name.replace("_", "-")](
107+
device="cuda"
108+
)
107109
return ORBCalculator(orbff, device="cuda")
108110

109111
def _init_sevennet_calculator(self) -> Calculator:

lambench/tasks/calculator/torsionnet/torsionnet.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@ def run_torsionnet(
2727
assert len(frame) == 1
2828
atoms: Atoms = frame.to_ase_structure()[0] # type: ignore
2929
atoms.calc = model.calc
30-
pred_energy = atoms.get_potential_energy() * 23.0609 # eV to kcal/mol
30+
pred_energy = atoms.get_potential_energy() * 23.0609 # eV to kcal/mol
3131
result[fragment.name].append(pred_energy)
3232
label_energy = frame.data["energies"][0]
3333
label[fragment.name].append(label_energy)

0 commit comments

Comments
 (0)