diff --git a/soda/dask/soda/data_sources/dask_cursor.py b/soda/dask/soda/data_sources/dask_cursor.py index ae302cb54..f143c4bf1 100644 --- a/soda/dask/soda/data_sources/dask_cursor.py +++ b/soda/dask/soda/data_sources/dask_cursor.py @@ -40,7 +40,7 @@ def fetchone(self) -> tuple: if self.df.empty: row_value = [] for col_dtype in self.df.dtypes: - if col_dtype in ["int", "float"]: + if np.issubdtype(col_dtype, np.number): row_value.append(0) else: row_value.append(None)