Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Nov 28, 2024
1 parent ce6a8bf commit b5ea252
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions tests/unit_tests/layers/test_properties.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,18 +13,18 @@ def setup_data():

g = jnp.ones((n_atoms, n_features))
R = jnp.array([[1.0, 0.0, 0.0], [0.0, 1.0, 0.0], [0.0, 0.0, 1.0], [1.0, 1.0, 0.0], [0.0, 1.0, 1.0]]) # atom positions
dr_vec = None
dr_vec = None
Z = jnp.array([1, 6, 8, 1, 6]) # atomic numbers
idx = None
box = None
idx = None
box = None

return g, R, dr_vec, Z, idx, box


def test_property_head(setup_data):
"""Test PropertyHead class functionality."""
g, R, dr_vec, Z, idx, box = setup_data

# Instantiate PropertyHead
property_head = PropertyHead(pname="property")

Expand Down

0 comments on commit b5ea252

Please sign in to comment.