maxframe 1.2.0__tar.gz → 1.2.1__tar.gz

{maxframe-1.2.0 → maxframe-1.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: maxframe
-Version: 1.2.0
+Version: 1.2.1
 Summary: MaxFrame operator-based data analyze framework
 Requires-Dist: numpy<2.0.0,>=1.19.0
 Requires-Dist: pandas>=1.0.0

{maxframe-1.2.0 → maxframe-1.2.1}/maxframe/dataframe/datasource/read_odps_query.py

@@ -262,7 +262,7 @@ class DataFrameReadODPSQuery(
     column_renames = DictField("column_renames", default=None)
 
     def get_columns(self):
-        return self.columns
+        return self.columns or list(self.dtypes.index)
 
     def set_pruned_columns(self, columns, *, keep_order=None):  # pragma: no cover
         self.columns = columns

maxframe-1.2.1/maxframe/lib/dtypes_extension/__init__.py

@@ -0,0 +1,14 @@
+# Copyright 1999-2025 Alibaba Group Holding Ltd.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+from .dtypes import ArrowDtype, dict_, is_map_dtype

{maxframe-1.2.0 → maxframe-1.2.1}/maxframe/lib/dtypes_extension/dtypes.py

@@ -13,10 +13,8 @@
 # limitations under the License.
 from typing import Union
 
-import numpy as np
 import pandas as pd
 import pyarrow as pa
-from pandas.api.extensions import ExtensionDtype
 
 try:
     from pandas import ArrowDtype
@@ -45,47 +43,3 @@ def is_map_dtype(dtype: ArrowDtype) -> bool:
     if ArrowDtype is None:
         raise ImportError("ArrowDtype is not supported in current environment")
     return isinstance(dtype, ArrowDtype) and isinstance(dtype.pyarrow_dtype, pa.MapType)
-
-
-_dtype_mapping = {
-    pd.Int8Dtype(): lambda x: pa.int8(),
-    pd.Int16Dtype(): lambda x: pa.int16(),
-    pd.Int32Dtype(): lambda x: pa.int32(),
-    pd.Int64Dtype(): lambda x: pa.int64(),
-    pd.UInt8Dtype(): lambda x: pa.uint8(),
-    pd.UInt16Dtype(): lambda x: pa.uint16(),
-    pd.UInt32Dtype(): lambda x: pa.uint32(),
-    pd.UInt64Dtype(): lambda x: pa.uint64(),
-    pd.BooleanDtype(): lambda x: pa.bool_(),
-    pd.Float32Dtype(): lambda x: pa.float32(),
-    pd.Float64Dtype(): lambda x: pa.float64(),
-    pd.StringDtype(): lambda x: pa.string(),
-}
-
-
-def infer_arrow_dtype(
-    dtype: Union[np.dtype, pa.DataType, ExtensionDtype]
-) -> Union[ArrowDtype, ExtensionDtype]:
-    """
-    Convert any pandas accepted dtype to arrow type in a best effort way.
-
-    Parameters
-    ----------
-    dtype : Union[np.dtype, pa.DataType, ExtensionDtype]
-        The dtype instance, can be np.dtype, pa.DataType or ExtensionDtype
-
-    Returns
-    -------
-    Union[pd.ArrowDtype, ExtensionDtype]: The converted pd.ArrowDtype, or ExtensionDtype if conversion failed.
-    """
-    if isinstance(dtype, ArrowDtype):
-        return dtype
-
-    if isinstance(dtype, np.dtype):
-        return ArrowDtype(pa.from_numpy_dtype(dtype))
-    if isinstance(dtype, pd.DatetimeTZDtype):
-        return pa.timestamp(dtype.unit, dtype.tz)
-
-    if dtype in _dtype_mapping:
-        return ArrowDtype(_dtype_mapping[dtype](dtype))
-    return dtype

{maxframe-1.2.0 → maxframe-1.2.1}/maxframe.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: maxframe
-Version: 1.2.0
+Version: 1.2.1
 Summary: MaxFrame operator-based data analyze framework
 Requires-Dist: numpy<2.0.0,>=1.19.0
 Requires-Dist: pandas>=1.0.0

maxframe-1.2.0/maxframe/lib/dtypes_extension/__init__.py

@@ -1,14 +0,0 @@
-# Copyright 1999-2025 Alibaba Group Holding Ltd.
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#      http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-from .dtypes import ArrowDtype, dict_, infer_arrow_dtype, is_map_dtype