Update analysis.py

alvaro-gc95 · Jul 12, 2024 · cad540c · cad540c
1 parent 8834896
commit cad540c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/rascal/analysis.py b/rascal/analysis.py
@@ -810,7 +810,7 @@ def clean_df(df):
     """
     assert isinstance(df, pd.DataFrame), "df needs to be a pd.DataFrame"
     df.dropna(inplace=True)
-    indices_to_keep = ~df.isin([np.nan, np.inf, -np.inf]).any(1)
+    indices_to_keep = ~df.isin([np.nan, np.inf, -np.inf]).any(axis=1)
     return df[indices_to_keep].astype(np.float64)