fix(eda): add geopoint in type count

sfu-db · Apr 28, 2021 · 94cbca2 · 94cbca2
1 parent 5974094
commit 94cbca2
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/dataprep/eda/dtypes.py b/dataprep/eda/dtypes.py
@@ -366,6 +366,8 @@ def get_dtype_cnts_and_num_cols(
             dtype_cnts["DateTime"] += 1
         elif is_dtype(col_dtype, GeoGraphy()):
             dtype_cnts["GeoGraphy"] += 1
+        elif is_dtype(col_dtype, GeoPoint()):
+            dtype_cnts["GeoPoint"] += 1
         else:
             raise NotImplementedError
     return dtype_cnts, num_cols