fix: removing deprecated np.float (#452)

lamalab-org · May 12, 2023 · 29ee1e0 · 29ee1e0
1 parent 37f246d
commit 29ee1e0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/mofdscribe/featurizers/chemistry/energygrid.py b/src/mofdscribe/featurizers/chemistry/energygrid.py
@@ -57,7 +57,7 @@ def read_ascii_grid(filename: Union[str, os.PathLike]) -> pd.DataFrame:
         names=["x", "y", "z", "energy", "deriv_x", "deriv_y", "deriv_z"],
         na_values="?",
     )
-    df = df.astype(np.float)
+    df = df.astype(float)
     return df
 
 

diff --git a/tests/featurizers/bu/test_rdkit_adaptor.py b/tests/featurizers/bu/test_rdkit_adaptor.py
@@ -12,7 +12,7 @@ def test_rdkit_adaptor(molecule_graph, molecule):
 
     features_a = adaptor.featurize(molecule)
     assert features_a.shape == (1,)
-    assert features_a.dtype == np.float
+    assert features_a.dtype == float
 
     features_b = adaptor.featurize(molecule_graph)
     assert all(features_a == features_b)