replace np.float with float

graphnet-team · Sep 22, 2023 · 6f014fa · 6f014fa
1 parent 8c54c77
commit 6f014fa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/graphnet/models/graphs/graph_definition.py b/src/graphnet/models/graphs/graph_definition.py
@@ -204,7 +204,7 @@ def _perturb_input(self, node_features: np.ndarray) -> np.ndarray:
             perturbed_features = self.rng.normal(
                 loc=node_features[:, self._perturbation_cols],
                 scale=np.array(
-                    list(self._perturbation_dict.values()), dtype=np.float
+                    list(self._perturbation_dict.values()), dtype=float
                 ),
             )
             node_features[:, self._perturbation_cols] = perturbed_features