MAINT: add support for Pandas 3.0rc

Author: gdementen

larray/core/array.py

@@ -6981,8 +6981,23 @@ def to_hdf(self, filepath, key) -> None:
         >>> a.to_hdf('test.h5', 'arrays/a')   # doctest: +SKIP
         """
         key = _translate_group_key_hdf(key)
+        def ensure_non_string_index(index):
+            if (isinstance(index, pd.MultiIndex) and
+                    any(isinstance(dt, pd.StringDtype) for dt in index.dtypes)):
+                obj_levels = [pd.Index(level_labels, dtype=object)
+                              for level_labels in index.levels]
+                index = pd.MultiIndex(obj_levels, index.codes, names=index.names)
+            return index
+
         with LHDFStore(filepath) as store:
-            store.put(key, self.to_frame())
+            df = self.to_frame()
+            # workaround for https://github.com/pandas-dev/pandas/issues/63412
+            # Since we created the dataframe ourselves, we can just
+            # replace the index inplace
+            df.index = ensure_non_string_index(df.index)
+            df.columns = ensure_non_string_index(df.columns)
+
+            store.put(key, df)
             attrs = store.get_storer(key).attrs
             attrs.type = 'Array'
             attrs.writer = 'LArray'

larray/core/group.py

@@ -591,7 +591,9 @@ def _to_key(v, stack_depth=1, parse_single_int=False) -> Key:
             return cls(key, name=name, axis=axis)
         else:
             return _seq_str_to_seq(v, stack_depth + 1, parse_single_int=parse_single_int)
-    elif v is Ellipsis or np.isscalar(v) or isinstance(v, (Group, slice, list, np.ndarray, ABCArray, OrderedSet)):
+    elif (v is Ellipsis or np.isscalar(v) or
+          isinstance(v, (Group, slice, list, np.ndarray, ABCArray, OrderedSet,
+                         pd.arrays.ArrowStringArray))):
         return v
     else:
         raise TypeError(f"{v} has an invalid type ({type(v).__name__}) for a key")