Reject all extension types as string types.

rapidsai · Mar 25, 2021 · babcdfc · babcdfc
1 parent 26bafd0
commit babcdfc
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/python/cudf/cudf/utils/dtypes.py b/python/cudf/cudf/utils/dtypes.py
@@ -155,9 +155,13 @@ def is_numerical_dtype(obj):
 
 def is_string_dtype(obj):
     return (
-        not is_decimal_dtype(obj)
-        and pd.api.types.is_string_dtype(obj)
+        pd.api.types.is_string_dtype(obj)
+        # Reject all cudf extension types.
         and not is_categorical_dtype(obj)
+        and not is_decimal_dtype(obj)
+        and not is_list_dtype(obj)
+        and not is_struct_dtype(obj)
+        and not is_interval_dtype(obj)
     )