np.unique -> set

Signed-off-by: Dmitry Chigarev <[email protected]>
modin-project · Nov 16, 2023 · 7469611 · 7469611
1 parent eeb7721
commit 7469611
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/modin/core/dataframe/pandas/metadata/dtypes.py b/modin/core/dataframe/pandas/metadata/dtypes.py
@@ -81,7 +81,7 @@ def __init__(
             )
         # columns with unknown dtypes (they're not described by 'remaining_dtype')
         if cols_with_unknown_dtypes is not None and len(
-            np.unique(cols_with_unknown_dtypes)
+            set(cols_with_unknown_dtypes)
         ) != len(cols_with_unknown_dtypes):
             raise NotImplementedError(
                 "Duplicated column names are not yet supported by DtypesDescriptor"
@@ -214,7 +214,7 @@ def lazy_get(
         DtypesDescriptor
             Descriptor that describes dtypes for columns specified in `ids`.
         """
-        if len(np.unique(ids)) != len(ids):
+        if len(set(ids)) != len(ids):
             raise NotImplementedError(
                 "Duplicated column names are not yet supported by DtypesDescriptor"
             )