Skip to content

Commit

Permalink
fixed linting
Browse files Browse the repository at this point in the history
  • Loading branch information
bowen-bd committed Mar 20, 2024
1 parent bcb71fd commit 340b9da
Show file tree
Hide file tree
Showing 2 changed files with 1 addition and 4 deletions.
1 change: 0 additions & 1 deletion src/matgl/layers/_norm.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,5 @@
"""Normalization modules."""


from __future__ import annotations

from typing import TYPE_CHECKING, Literal
Expand Down
4 changes: 1 addition & 3 deletions tests/ext/test_ase.py
Original file line number Diff line number Diff line change
Expand Up @@ -44,9 +44,7 @@ def test_PESCalculator_and_M3GNetCalculator(MoS):
def test_CHGNetCalculator(MoS):
adaptor = AseAtomsAdaptor()
s_ase = adaptor.get_atoms(MoS) # type: ignore
ff = load_model(
"pretrained_models/CHGNet-MPtrj-2023.12.1-PES-2.7M/"
)
ff = load_model("pretrained_models/CHGNet-MPtrj-2023.12.1-PES-2.7M/")
ff.calc_hessian = True
calc = PESCalculator(potential=ff)
s_ase.set_calculator(calc)
Expand Down

0 comments on commit 340b9da

Please sign in to comment.