[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
apax-hub · Jan 20, 2025 · 828f1bb · 828f1bb
1 parent d113b69
commit 828f1bb
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/apax/nn/builder.py b/apax/nn/builder.py
@@ -95,7 +95,7 @@ def build_readout(self, head_config, is_feature_fn=False, only_use_n_layers=None
             dtype = head_config["dtype"]
         else:
             raise KeyError("No dtype specified in config")
-        
+
         nn_layers = head_config["nn"]
         if only_use_n_layers is not None:
             nn_layers = nn_layers[:only_use_n_layers]

diff --git a/tests/conftest.py b/tests/conftest.py
@@ -79,7 +79,7 @@ def example_atoms_list(num_data: int, pbc: bool, calc_results: List[str]) -> Ato
     for _ in range(num_data):
         num_atoms = np.random.randint(10, 15)
         atoms = create_example_atoms(num_atoms, pbc, calc_results)
-        
+
         atoms_list.append(atoms)
 
     return atoms_list

diff --git a/tests/unit_tests/nn/test_builder.py b/tests/unit_tests/nn/test_builder.py
@@ -18,7 +18,7 @@ def test_builder_feature_model():
     model = builder.build_energy_model()
     key = jax.random.PRNGKey(0)
     params = model.init(key, R, Z, idx, box, offsets)
-    
+
     model = builder.build_feature_model(only_use_n_layers=0, init_box=box)
     out = model.apply(params, R, Z, idx, box, offsets)
     assert out.shape == (360,)