Simplify hasattr at count_nonzero function from array module

geopozo · Jun 6, 2024 · cda1a2d · cda1a2d
1 parent c2ce01f
commit cda1a2d
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/pozo/utils/array.py b/pozo/utils/array.py
@@ -304,10 +304,7 @@ def isnan(data):
 def count_nonzero(data):
     if hasattr(data, "coords"):  # xarray
         return np.count_nonzero(data.values)
-    elif hasattr(data, "isnull"):  # pandas
-        return np.count_nonzero(data.to_numpy())
-    elif hasattr(data, "is_null"):  # polars
-        print(np.count_nonzero(data.to_numpy()))
+    elif hasattr(data, "to_numpy"):  # series
         return np.count_nonzero(data.to_numpy())
     else:
         check_numpy()