Pythonic object name set uniqueness

rapidsai · May 3, 2024 · 2124759 · 2124759
1 parent b5de816
commit 2124759
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/python/cudf/cudf/core/reshape.py b/python/cudf/cudf/core/reshape.py
@@ -419,10 +419,7 @@ def concat(objs, axis=0, join="outer", ignore_index=False, sort=None):
         else:
             # All levels in the multiindex label must have the same type
             has_multiple_level_types = (
-                len(
-                    set().union(*(map(type, obj._data.keys()) for obj in objs))
-                )
-                > 1
+                len({type(name) for o in objs for name in o._data.keys()}) > 1
             )
             if has_multiple_level_types:
                 raise NotImplementedError(