Skip to content

Commit

Permalink
formatting
Browse files Browse the repository at this point in the history
  • Loading branch information
bananenpampe committed Nov 26, 2024
1 parent c8803aa commit 7d1c88d
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 1 deletion.
4 changes: 3 additions & 1 deletion src/shiftml/ase/calculator.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,9 @@
}

resolve_outputs = {
"ShiftML1.1rev": {"mtt::cs_iso": ModelOutput(quantity="", unit="ppm", per_atom=True)},
"ShiftML1.1rev": {
"mtt::cs_iso": ModelOutput(quantity="", unit="ppm", per_atom=True)
},
"ShiftML2.0rev": {
"mtt::cs_iso": ModelOutput(quantity="", unit="ppm", per_atom=True),
"mtt::cs_iso_std": ModelOutput(quantity="", unit="ppm", per_atom=True),
Expand Down
1 change: 1 addition & 0 deletions tests/test_regression_pretrained.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,7 @@
"ShiftML1.1rev": np.array([137.5415, 137.5415]),
}


def test_shiftml1_1_regression():
"""Regression test for the ShiftML1.1 model."""

Expand Down

0 comments on commit 7d1c88d

Please sign in to comment.