Merge pull request #49 from intake/chuyuanliu/vectorize-concatenation…

…-in-pandas fix: avoid probable to_list/from_iter in pandas.concat
intake · Mar 10, 2024 · f8bfbf3 · f8bfbf3
2 parents d0f7893 + 2779db0
commit f8bfbf3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/awkward_pandas/__init__.py b/src/awkward_pandas/__init__.py
@@ -1,6 +1,6 @@
 from __future__ import annotations
 
-import awkward_pandas.accessor  # noqa
+import awkward_pandas.accessor
 import awkward_pandas.dask_connect  # noqa
 from awkward_pandas.array import AwkwardExtensionArray
 from awkward_pandas.dtype import AwkwardDtype

diff --git a/src/awkward_pandas/array.py b/src/awkward_pandas/array.py
@@ -129,7 +129,7 @@ def copy(self):
 
     @classmethod
     def _concat_same_type(cls, to_concat):
-        return cls(ak.concatenate(to_concat))
+        return cls(ak.concatenate([a._data for a in to_concat]))
 
     @property
     def ndim(self) -> Literal[1]: