maxframe 0.1.0b1__cp38-cp38-win_amd64.whl → 0.1.0b3__cp38-cp38-win_amd64.whl

maxframe/udf.py

@@ -26,9 +26,7 @@ from .serialization.serializables import (
 
 class MarkedFunction(Serializable):
     func = FunctionField("func")
-    resource_libraries = ListField(
-        "resource_libraries", FieldTypes.string, default_factory=list
-    )
+    resources = ListField("resources", FieldTypes.string, default_factory=list)
 
     def __init__(self, func: Optional[Callable] = None, **kw):
         super().__init__(func=func, **kw)
@@ -43,33 +41,34 @@ class MarkedFunction(Serializable):
         return f"<MarkedFunction {self.func!r}>"
 
 
-def with_resource_libraries(
-    *resources: Union[str, Resource], use_wrapper_class: bool = True
-):
+def with_resources(*resources: Union[str, Resource], use_wrapper_class: bool = True):
     def res_to_str(res: Union[str, Resource]) -> str:
         if isinstance(res, str):
             return res
-        res_parts = [res.project]
+        res_parts = [res.project.name]
         if res.schema:
-            res_parts.append(res.schema)
-        res_parts.append(res.name)
-        return ".".join(res_parts)
+            res_parts.extend(["schemas", res.schema])
+        res_parts.extend(["resources", res.name])
+        return "/".join(res_parts)
 
     def func_wrapper(func):
         str_resources = [res_to_str(r) for r in resources]
         if not use_wrapper_class:
-            func.resource_libraries = str_resources
+            func.resources = str_resources
             return func
 
         if isinstance(func, MarkedFunction):
-            func.resource_libraries = str_resources
+            func.resources = str_resources
             return func
-        return MarkedFunction(func, resource_libraries=list(resources))
+        return MarkedFunction(func, resources=list(str_resources))
 
     return func_wrapper
 
 
-def get_udf_resource_libraries(
+with_resource_libraries = with_resources
+
+
+def get_udf_resources(
     func: Callable,
-) -> Optional[List[Union[Resource, str]]]:
-    return getattr(func, "resource_libraries", None)
+) -> List[Union[Resource, str]]:
+    return getattr(func, "resources", None) or []

maxframe/utils.py

@@ -338,6 +338,14 @@ def deserialize_serializable(ser_serializable: bytes):
     return deserialize(header2, buffers2)
 
 
+def skip_na_call(func: Callable):
+    @functools.wraps(func)
+    def new_func(x):
+        return func(x) if x is not None else None
+
+    return new_func
+
+
 def url_path_join(*pieces):
     """Join components of url into a relative url
 
@@ -450,6 +458,9 @@ _ToThreadRetType = TypeVar("_ToThreadRetType")
 
 
 class ToThreadMixin:
+    _thread_pool_size = 1
+    _counter = itertools.count().__next__
+
     def __del__(self):
         if hasattr(self, "_pool"):
             kw = {"wait": False}
@@ -466,7 +477,10 @@ class ToThreadMixin:
         **kwargs,
     ) -> _ToThreadRetType:
         if not hasattr(self, "_pool"):
-            self._pool = concurrent.futures.ThreadPoolExecutor(1)
+            self._pool = concurrent.futures.ThreadPoolExecutor(
+                self._thread_pool_size,
+                thread_name_prefix=f"{type(self).__name__}Pool-{self._counter()}",
+            )
 
         task = asyncio.create_task(
             to_thread_pool(func, *args, **kwargs, pool=self._pool)
@@ -1081,3 +1095,9 @@ def arrow_type_from_str(type_str: str) -> pa.DataType:
     if len(value_stack) > 1:
         raise ValueError(f"Cannot parse type {type_str}")
     return value_stack[-1]
+
+
+def get_python_tag():
+    # todo add implementation suffix for non-GIL tags when PEP703 is ready
+    version_info = sys.version_info
+    return f"cp{version_info[0]}{version_info[1]}"

{maxframe-0.1.0b1.dist-info → maxframe-0.1.0b3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: maxframe
-Version: 0.1.0b1
+Version: 0.1.0b3
 Summary: MaxFrame operator-based data analyze framework
 Requires-Dist: numpy >=1.19.0
 Requires-Dist: pandas >=1.0.0
@@ -102,76 +102,3 @@ License
 
 Licensed under the `Apache License
 2.0 <https://www.apache.org/licenses/LICENSE-2.0.html>`__.
-MaxCompute MaxFrame Client
-==========================
-
-MaxFrame is a computational framework created by Alibaba Cloud to
-provide a way for Python developers to parallelize their code with
-MaxCompute. It creates a runnable computation graph locally, submits it
-to MaxCompute to execute and obtains results from MaxCompute.
-
-MaxFrame client is the client of MaxFrame. Currently it provides a
-DataFrame-based SDK with compatible APIs for pandas. In future, other
-common Python libraries like numpy and scikit-learn will be added as
-well. Python 3.7 is recommended for MaxFrame client to enable all
-functionalities while supports for higher Python versions are on the
-way.
-
-Installation
-------------
-
-You may install MaxFrame client through PIP:
-
-.. code:: bash
-
-   pip install maxframe
-
-Latest beta version can be installed with ``--pre`` argument:
-
-.. code:: bash
-
-   pip install --pre maxframe
-
-You can also install MaxFrame client from source code:
-
-.. code:: bash
-
-   pip install git+https://github.com/aliyun/aliyun-odps-maxframe.git
-
-Getting started
----------------
-
-We show a simple code example of MaxFrame client which read data from a
-MaxCompute table, performs some simple data transform and writes back
-into MaxCompute.
-
-.. code:: python
-
-   import maxframe.dataframe as md
-   import os
-   from maxframe import new_session
-   from odps import ODPS
-
-   o = ODPS(
-       os.getenv('ALIBABA_CLOUD_ACCESS_KEY_ID'),
-       os.getenv('ALIBABA_CLOUD_ACCESS_KEY_SECRET'),
-       project='your-default-project',
-       endpoint='your-end-point',
-   )
-   session = new_session(o)
-
-   df = md.read_odps_table("source_table")
-   df["A"] = "prefix_" + df["A"]
-   md.to_odps_table(df, "prefix_source_table")
-
-Documentation
--------------
-
-Detailed documentations can be found
-`here <https://maxframe.readthedocs.io>`__.
-
-License
--------
-
-Licensed under the `Apache License
-2.0 <https://www.apache.org/licenses/LICENSE-2.0.html>`__.

{maxframe-0.1.0b1.dist-info → maxframe-0.1.0b3.dist-info}/RECORD

@@ -1,20 +1,21 @@
 maxframe/__init__.py,sha256=MgltwhBvnUQDKKmHDg9Y69TJkRGmQQ9m8-D9FC2pcLU,1007
-maxframe/_utils.cp38-win_amd64.pyd,sha256=ycV3O8A50cR123JdJwawsyhchDobNV2qpdMZ2eSel-8,308224
+maxframe/_utils.cp38-win_amd64.pyd,sha256=v0SiHG8U69LbmO5jg_wMpsYNpTlx3bjMIsvfOeEDnxE,308736
 maxframe/_utils.pxd,sha256=_qHN-lCY1FQgDFIrrqA79Ys0SBdonp9kXRMS93xKSYk,1187
 maxframe/_utils.pyx,sha256=_3p6aJEJ6WZYLcNZ6o4DxoxsxqadTlJXFlgDeFPxqUQ,17564
-maxframe/codegen.py,sha256=LzOBM2H5U24Eis3NA9tiy5dNe5xKsk7YaUizZKkkG44,14020
+maxframe/codegen.py,sha256=S23hTTu2fIJTv5PMIItE_NPUnadUyoiMmb-1YAqbaWw,16329
 maxframe/conftest.py,sha256=JE9I-5mP4u-vgUqYL22mNY3tqpGofM8VMe8c8VUYkzk,4403
 maxframe/env.py,sha256=xY4wjMWIJ4qLsFAQ5F-X5CrVR7dDSWiryPXni0YSK5c,1435
+maxframe/errors.py,sha256=xBnvoJjjNcHVLhwj77Dux9ut8isGVmmJXFqefmmx8Ak,711
 maxframe/extension.py,sha256=o7yiS99LWTtLF7ZX6F78UUJAqUyd-LllOXA2l69np50,2455
 maxframe/mixin.py,sha256=QfX0KqVIWDlVDSFs0lwdzLexw7lS7W_IUuK7aY1Ib8c,3624
-maxframe/opcodes.py,sha256=Kj8JQaGtnGXvcElBlrW1lZriSM9hCQRXUfJ5lic3pGA,10541
-maxframe/protocol.py,sha256=WsirflXFUuorZIQTddGbE1rFGP-5wR4RwcIhIKHqXow,13768
+maxframe/opcodes.py,sha256=L-BvSFEUOMrtGJFXIH6zk2Xv_-te5VZxv5eDDCNNt0U,10566
+maxframe/protocol.py,sha256=N4i0ggLY131gwnxOrCgKeZwzhLKSRB171cx1lWRvUcw,14605
 maxframe/session.py,sha256=Mme-jB2hioJC6ttrXfX4XSeHLCMYoFR0ikmmhx82-vc,36624
 maxframe/typing_.py,sha256=pAgOhHHSM376N7PJLtNXvS5LHNYywz5dIjnA_hHRWSM,1133
-maxframe/udf.py,sha256=23Zd-abK40xMgT3sS7aCn3G8f-0t1mcTy23M0MdatmE,2332
-maxframe/utils.py,sha256=c_xOE-9ecd7kiDmuCub-bpHFtB1-1dzzpKtoCUlMIuc,34452
+maxframe/udf.py,sha256=EFAAV2c8SpWKcF9_8Pocpjc4bXsEASf57Qy_Q30YH4Q,2315
+maxframe/utils.py,sha256=CpA4Cqf5Lg7LMKJeJFsmybqVutcSh3sSqMhdHnIi0h4,35017
 maxframe/config/__init__.py,sha256=AHo3deaCm1JnbbRX_udboJEDYrYytdvivp9RFxJcumI,671
-maxframe/config/config.py,sha256=9FrjTv-Dx-DIBW0FejJ86CKVRx39Ycj2EVIlstzLjxo,13178
+maxframe/config/config.py,sha256=Top-9hUEkPvaTlAGrP9GzGjwvZk6YPuHtwy4TEryKqg,13475
 maxframe/config/validators.py,sha256=pKnloh2kEOBRSsT8ks-zL8XVSaMMVIEvHvwNJlideeo,1672
 maxframe/config/tests/__init__.py,sha256=29eM5D4knhYwe3TF42naTuC5b4Ym3VeH4rK8KpdLWNY,609
 maxframe/config/tests/test_config.py,sha256=FWQZ6KBUG_jY1-KaR-GKXl7khhlTbuLlk3uaEV8koM8,2839
@@ -25,15 +26,15 @@ maxframe/core/mode.py,sha256=a-2AjLrIaqemN3pZPFhdfrPYXR7ryCLcsT1KJwWKPb0,3107
 maxframe/core/entity/__init__.py,sha256=tD4zo3KXpzLrQraHnIXeO1Q961lSsIqpbAGRK2WijVE,1336
 maxframe/core/entity/chunks.py,sha256=zKk8Iyc3IkakIDW1bMYq_zZNLrR4ZMdXH-mBuOiFerM,2202
 maxframe/core/entity/core.py,sha256=aFwjNMhTJ4ybr1WzmMVSTG211fzutzaATs14QoNh-JM,4170
-maxframe/core/entity/executable.py,sha256=Pt-ad4u2jqb9mCbOHp0DcL1Oo6kjUzF411ZxQrouqys,11239
+maxframe/core/entity/executable.py,sha256=CKxFGvFPfY_8JBprhpyndhTSLgVLtUG4G5n7Dw0dHnw,11275
 maxframe/core/entity/fuse.py,sha256=X1lI0WXj5t0flgGI5-qlVl5LoYkAdLJHk2Vv767C9G4,2350
-maxframe/core/entity/objects.py,sha256=GST_N6nwNxUv1wXaPBWty7n8Asn2hX8wfIDjrctkZP8,3119
+maxframe/core/entity/objects.py,sha256=Ys_l6cBp0HwgRmXuqYo4HsnjdbfUW4mgvek5W0IMmXY,3134
 maxframe/core/entity/output_types.py,sha256=NnNeDBVAEhD8dtPBWzpM7n6s8neVFrahjd0zMGWroCc,2735
 maxframe/core/entity/tileables.py,sha256=6jJyFscvb8sH5K_k2VaNGeUm8YrpevCtou3WSUl4Dw8,13973
 maxframe/core/entity/utils.py,sha256=454RYVbTMVW_8KnfDqUPec4kz1p98izVTC2OrzhOkao,966
 maxframe/core/graph/__init__.py,sha256=n1WiszgVu0VdXsk12oiAyggduNwu-1-9YKnfZqvmmXk,838
-maxframe/core/graph/core.cp38-win_amd64.pyd,sha256=iuBSHogC_TrMCtP9IBj2KUyRXM8SJCvuF_A9D9gE3Ng,250368
-maxframe/core/graph/core.pyx,sha256=tHD1r5h2MoqeqtJWZIo_wkdz3WIK0yWRInRj-Aendkc,16320
+maxframe/core/graph/core.cp38-win_amd64.pyd,sha256=Pxd3Gfcq9JBPGAfySG6YaTFnJiTphRtr1FNe3NZCBUo,250368
+maxframe/core/graph/core.pyx,sha256=WYlYtXXSs72vfhf2ttJO-4u85exYzy2J9mlALHOMqoA,16354
 maxframe/core/graph/entity.py,sha256=RT_xbP5niUN5D6gqZ5Pg1vUegHn8bqPk8G8A30quOVA,5730
 maxframe/core/graph/builder/__init__.py,sha256=vTRY5xRPOMHUsK0jAtNIb1BjSPGqi_6lv86AroiiiL4,718
 maxframe/core/graph/builder/base.py,sha256=IzNYD4A2alnYaOvZnipDzC-ZIfNsYR80wLnFcx7bJzc,2635
@@ -43,7 +44,7 @@ maxframe/core/graph/builder/utils.py,sha256=vO7_4ykzI1G194Oqn_tM6zVWwxLReJ0GL4lC
 maxframe/core/graph/tests/__init__.py,sha256=29eM5D4knhYwe3TF42naTuC5b4Ym3VeH4rK8KpdLWNY,609
 maxframe/core/graph/tests/test_graph.py,sha256=nkVn_dLk0oT42HWJxnSjniC5ElXjsHxC1Jzlg1CKHcQ,7667
 maxframe/core/operator/__init__.py,sha256=IHBCB3KC6dlVnfJryHSd__3Xnso9CtwYJrbfsSJ6siE,1162
-maxframe/core/operator/base.py,sha256=ops_YDdVqJtGSu05E4efQqGbxCkhztohn_sAxjpvwJI,15277
+maxframe/core/operator/base.py,sha256=3d2_iNxp06XCiiY_zZRNxm2Cvy2kR-5x9Di6tqbuuxA,15701
 maxframe/core/operator/core.py,sha256=cDzq0_9k7pfUiyAkaIq2eLaP6fAw7uRS-7bacon0Acg,9539
 maxframe/core/operator/fetch.py,sha256=4g9kyJbIco7cDgbdUJOJNtC3NCploUU45jnateGYG8c,1563
 maxframe/core/operator/fuse.py,sha256=1XBA2T6X4WSMj8xssZoC8bSALuNIdYSg7NTKTgR14-E,954
@@ -53,7 +54,7 @@ maxframe/core/operator/tests/__init__.py,sha256=29eM5D4knhYwe3TF42naTuC5b4Ym3VeH
 maxframe/core/operator/tests/test_core.py,sha256=iqZk4AWubFLO24V_VeV6SEy5xrzBFLP9qKK6tKO0SGs,1755
 maxframe/core/tests/__init__.py,sha256=29eM5D4knhYwe3TF42naTuC5b4Ym3VeH4rK8KpdLWNY,609
 maxframe/core/tests/test_mode.py,sha256=fyRH-ksa6MogEs6kNhtXhCZyvhYqflgaXJYI3nSo-ps,2507
-maxframe/dataframe/__init__.py,sha256=2YCnRh0neFIiFhmRr0wm3BHPRsvhEMjLXxs7NPY90PM,2109
+maxframe/dataframe/__init__.py,sha256=t_Fnu2MNCgOr_N7mMqGU3LeZXvBF1yOHtLb5u1PoHt0,2187
 maxframe/dataframe/arrays.py,sha256=rOvhxMQars9E3SOYSu0ygBuuRVY0QV6xzengnMqKs4s,29616
 maxframe/dataframe/core.py,sha256=63TqgmOCTr5wtwbILJ6bLtHZQ6xnZd7LvVrMqR27GDQ,76034
 maxframe/dataframe/initializer.py,sha256=WW96yQjquofNFt6RPZvgWW4SBmH0OEDj8-BxpuyKThY,10552
@@ -114,15 +115,16 @@ maxframe/dataframe/datasource/core.py,sha256=1X32MxCU3Y5vxTVEIkNNXx29eIARyRCvnkV
 maxframe/dataframe/datasource/dataframe.py,sha256=P3kvDS_Jh6EGgJRVfh_rHVkVfMaHeGQ2U7rZT5Gr2Tk,2082
 maxframe/dataframe/datasource/date_range.py,sha256=At-TQwqMiTEshSk4gQvxpq0wSEibC0FwpMDnwcjaI4o,17748
 maxframe/dataframe/datasource/from_index.py,sha256=c6ecX7miuig3-uZB3-aj68TkpUypzpaogZsUZSbTtBs,1911
-maxframe/dataframe/datasource/from_records.py,sha256=5uIi6GsbRl3aTx0CI_CFoXrAjz-bmt4_VfzIhBeo30M,3478
+maxframe/dataframe/datasource/from_records.py,sha256=ygpKOMXZnDdWzGxMxQ4KdGv-tJFT_yhpLGnDlS1rNLc,3549
 maxframe/dataframe/datasource/from_tensor.py,sha256=mShHYi0fZcG7ZShFVgIezaphh8tSFqR9-nQMm5YKIhw,15146
 maxframe/dataframe/datasource/index.py,sha256=X_NShW67nYJGxaWp3qOrvyInNkz9L-XHjbApU4fHoes,4518
 maxframe/dataframe/datasource/read_csv.py,sha256=IvQihmpcZIdzSD7ziX92aTAHNyP5WnTgd2cZz_h43sQ,24668
-maxframe/dataframe/datasource/read_odps_table.py,sha256=kZDpJvq3d39hLYa4oznk2LjJXipOUkfXadDhxPFSZbs,9267
+maxframe/dataframe/datasource/read_odps_query.py,sha256=QKU_7R6dd6GXzNX4IN9uwu3jB17MfhTr2nUtx0LFSNk,10116
+maxframe/dataframe/datasource/read_odps_table.py,sha256=ocSKQQ7SwIkzliFCFWNzy3e8J3CBZsf4971oRdlgdks,9294
 maxframe/dataframe/datasource/read_parquet.py,sha256=SZPrWoax2mwMBNvRk_3lkS72pZLe-_X_GwQ1JROBMs4,14952
 maxframe/dataframe/datasource/series.py,sha256=elQVupKETh-hUHI2fTu8TRxBE729Vyrmpjx17XlRV-8,1964
 maxframe/dataframe/datasource/tests/__init__.py,sha256=29eM5D4knhYwe3TF42naTuC5b4Ym3VeH4rK8KpdLWNY,609
-maxframe/dataframe/datasource/tests/test_datasource.py,sha256=jorppnuuEZMdCx_VhQu-HxlyBUiDj3r-Y8nIlmg02fs,11645
+maxframe/dataframe/datasource/tests/test_datasource.py,sha256=UumRBjE-bIuCi7Z4_3t8qb58ZcF8ePRZf3xF7DTvqIA,15041
 maxframe/dataframe/datastore/__init__.py,sha256=MmlHYvFacMReOHDQMXF-z2bCsLyrSHYBVwIlCsZGOK4,810
 maxframe/dataframe/datastore/to_csv.py,sha256=lheaF3ZmBPrcwcWyhK5gEVAGIaLJbvTyVAzqZFGG7eM,8026
 maxframe/dataframe/datastore/to_odps.py,sha256=Ml_iF9AspqIgGbeOAjTF3ukAwv-7SizribuqXZdxuXo,5776
@@ -133,10 +135,10 @@ maxframe/dataframe/extensions/tests/__init__.py,sha256=29eM5D4knhYwe3TF42naTuC5b
 maxframe/dataframe/extensions/tests/test_extensions.py,sha256=oDnVwQx-o8p4wmen8ZS3gnudOAihFAdNKQhSCqNNXzQ,1324
 maxframe/dataframe/fetch/__init__.py,sha256=W1arTCAjD0v_bdVeIzpJMnil3h0Ucsn29zFWmgZcYck,668
 maxframe/dataframe/fetch/core.py,sha256=27VANjpMm2rdCg1KPZxWZrKWNuNgSMzZrordI05mqWc,3424
-maxframe/dataframe/groupby/__init__.py,sha256=2GpnBNlM35Z2cU-seUkwZ_GX1h9vpiZXtX4HOo2y4zc,3325
+maxframe/dataframe/groupby/__init__.py,sha256=wMjmvk4ced1uCm7bw0oodIKvaep61KhupriL9JRRq5w,3443
 maxframe/dataframe/groupby/aggregation.py,sha256=cUnu-Bj6YD1TVkaafwL2aGIIqixLEq7s9-7BQ_1T2DI,12303
 maxframe/dataframe/groupby/apply.py,sha256=DQHyEfqj-3tfK-CxwpdVgya0_YC9dImeWYPZJDw7ckk,9735
-maxframe/dataframe/groupby/core.py,sha256=BPcDrMN710phqwV1_F7kZKKan7ZqxoER-Jd4mXxMMyw,6154
+maxframe/dataframe/groupby/core.py,sha256=NG6e3sqIu5dnBw9_DCQEDtsnxM5e4Yl1oD7Z_qjdtWA,6254
 maxframe/dataframe/groupby/cum.py,sha256=A7vIWLsb50VLu3yAngO-BfZecjWj0Fk6TZ5v4uQEAPM,3879
 maxframe/dataframe/groupby/fill.py,sha256=AXRmA_j-m7ig0udLCJ02FwIce2GLQ2U8KlnuCe-NY3U,4862
 maxframe/dataframe/groupby/getitem.py,sha256=owNzoE8UEfM41dfuntKkRBjjYYbY8O8CMJchIhCEyds,3344
@@ -197,7 +199,7 @@ maxframe/dataframe/misc/select_dtypes.py,sha256=qsrWW8BNBd-hAT1yIlrnbvBjfbzZyttz
 maxframe/dataframe/misc/shift.py,sha256=HCGKBuToksgu5LZR_k5eaQrdyrXCeqRZYbZs0J5-N6s,9191
 maxframe/dataframe/misc/stack.py,sha256=a3eWRUnFQ4okGUEirsdIq9taLfcRTCKnLfcXHg-sce8,8200
 maxframe/dataframe/misc/string_.py,sha256=AaEfHcPAJWZ8Ug-zbD0gjaBfuFXWfHDXbrHQ1Kaqes4,8322
-maxframe/dataframe/misc/to_numeric.py,sha256=0U96l_IxGHPzB3Tbomc_VXEJ04fYL43i0hgJ_0EqNLY,6370
+maxframe/dataframe/misc/to_numeric.py,sha256=VxYijzFTiUD9Jc62fE24tZ3iB8EVb9cBqZwvQFt2bHA,6440
 maxframe/dataframe/misc/transform.py,sha256=JotPUQzKGhCLRi53sk7USU9HscNjUKCzzus9PgTyhHU,12000
 maxframe/dataframe/misc/transpose.py,sha256=SsKRuN9Pj36D6kntnsLfzBsFHjSCiV1cZuPJuKHgbGU,3772
 maxframe/dataframe/misc/value_counts.py,sha256=7Yd3ZSrCRDMRX093IlzdsrTJ5UUx0n_lbD5AmXLUr_U,5674
@@ -262,13 +264,13 @@ maxframe/dataframe/ufunc/__init__.py,sha256=0oKXArwhdI6TvyMamPN_UuW7dwe9plKXAYJX
 maxframe/dataframe/ufunc/tensor.py,sha256=wZzmrGgGZ1w2EaYjjfc-J6vKe6W8SLP9yFNGUtMLXSc,1672
 maxframe/dataframe/ufunc/ufunc.py,sha256=lvHUhekjNn8eJZRfJRwk922j6QLQvIfrNWs41hJ-Vq0,1703
 maxframe/dataframe/window/__init__.py,sha256=bQsgFdk801ezvU3Z4i-V-JGIOX-JzwsQk2JPrIdgPUc,939
-maxframe/dataframe/window/aggregation.py,sha256=OUQDtCWlfShFaEUaQhyqMAvSJIBnJkDScwiJDJMYavE,4571
+maxframe/dataframe/window/aggregation.py,sha256=gTkI4ysG3Qd-pHiTI-aUWnsaGMZuk_gUhD8pZ-tZYko,3890
 maxframe/dataframe/window/core.py,sha256=NKWQIao5dOI_3jL2Rzkgo_wdi36VCBIMWG4eRe1237M,2274
-maxframe/dataframe/window/ewm.py,sha256=t7NtbcIrgOXCKannKrCYBYIXes5wBImCAsPYiW6C8bk,8288
+maxframe/dataframe/window/ewm.py,sha256=csPVRhHOTk2VP_I7WOuQMG6vlZtPXSij6A3Pe7H-who,8038
 maxframe/dataframe/window/expanding.py,sha256=Y0RLJdV4lnVGDD2VdOVCuYoXVr5PO-WqaMx3AAiNKWc,3941
 maxframe/dataframe/window/rolling.py,sha256=czrpdXbenldL-9tYc8ndRBf7KlzIIfP4jwmM5V-fwO8,12418
 maxframe/dataframe/window/tests/__init__.py,sha256=29eM5D4knhYwe3TF42naTuC5b4Ym3VeH4rK8KpdLWNY,609
-maxframe/dataframe/window/tests/test_ewm.py,sha256=4iFXBalVZuKT99AfuM9tU00Z0Z41iWiD8GEjkXLCDtE,2330
+maxframe/dataframe/window/tests/test_ewm.py,sha256=v1hM2ucFH_8aRn8AJqWEpvvVtSK9QQUCs911N0trbX8,2130
 maxframe/dataframe/window/tests/test_expanding.py,sha256=PsFYI6YXxNHYlsyaesy9yIAUSixp4IOrChmQtRGBpKc,2010
 maxframe/dataframe/window/tests/test_rolling.py,sha256=eJYHh4MhKm_e9h4bCnK_d_aAYlKGQtFAbZTTEILe6gU,1771
 maxframe/learn/__init__.py,sha256=1QzrFCJmdZFy0Nh1Ng3V6yY_NVvoSUdTIqcU-HIa1wk,649
@@ -281,7 +283,7 @@ maxframe/learn/contrib/pytorch/tests/test_pytorch.py,sha256=GHP-oD5uMU8LD90Jt2cH
 maxframe/lib/__init__.py,sha256=_PB28W40qku6YiT8fJYqdmEdRMQfelOwGeksCOZJfCc,657
 maxframe/lib/compression.py,sha256=QQpNK79iUC9zck74I0HKMhapSRnLBXtTRyS91taEVIc,1497
 maxframe/lib/functools_compat.py,sha256=2LTrkSw5i-z5E9XCtZzfg9-0vPrYxicKvDjnnNrAL1Q,2697
-maxframe/lib/mmh3.cp38-win_amd64.pyd,sha256=pFpTjj4_RHguFOFsaudsDifXrhrEEnQLV4sxOdaPj8k,17408
+maxframe/lib/mmh3.cp38-win_amd64.pyd,sha256=HMgH8mFYmm50QB_5WtKQ7UIOgGnFUGJoPzGKKfApJVs,17408
 maxframe/lib/version.py,sha256=VOVZu3KHS53YUsb_vQsT7AyHwcCWAgc-3bBqV5ANcbQ,18941
 maxframe/lib/wrapped_pickle.py,sha256=bzEaokhAZlkjXqw1xfeKO1KX2awhKIz_1RT81yPPoag,3949
 maxframe/lib/aio/__init__.py,sha256=xzIYnV42_7CYuDTTv8svscIXQeJMF0nn8AXMbpv173M,963
@@ -289,7 +291,7 @@ maxframe/lib/aio/_runners.py,sha256=zhDC92KxrYxLEufo5Hk8QU-mTVOxNL7IM9pZXas_nDg,
 maxframe/lib/aio/_threads.py,sha256=cDaEKg5STncq9QTPUUwehJ722vgueqBoB1C-NeoHN8E,1363
 maxframe/lib/aio/base.py,sha256=Ol0MnkcsBRfsQdZWceYfaWVtNOuiHzY8EYo2Zh0QFvM,2240
 maxframe/lib/aio/file.py,sha256=uy2LM_U8-Snpf45yZqUQRR_0hZT5UXZnwq0qENpMI6k,2097
-maxframe/lib/aio/isolation.py,sha256=zUqytRXUHRniUrZCRtsJkBldP-RnGWv3KpelQ3hyRdU,2723
+maxframe/lib/aio/isolation.py,sha256=wIliQ1qFtGV_cZ4stE21QHKUVtq7j88m7ZuZF8Ye2iE,2861
 maxframe/lib/aio/lru.py,sha256=hZ0QY8VWhZr06B11YqjEONKcjySP7oKaa-p9evwnxZY,7133
 maxframe/lib/aio/parallelism.py,sha256=Q3dXir6wr5vG2SmVSz0n6BdH7d5mhMTohfeFs5JDTtU,1272
 maxframe/lib/aio/tests/__init__.py,sha256=29eM5D4knhYwe3TF42naTuC5b4Ym3VeH4rK8KpdLWNY,609
@@ -344,17 +346,17 @@ maxframe/remote/core.py,sha256=w_eTDEs0O7iIzLn1YrMGh2gcNAzzbqV0mx2bRT7su_U,7001
 maxframe/remote/run_script.py,sha256=k93-vaFLUanWoBRai4-78DX_SLeZ8_rbbxcCtOIXZO8,3677
 maxframe/serialization/__init__.py,sha256=nxxU7CI6MRcL3sjA1KmLkpTGKA3KG30FKl-MJJ0MCdI,947
 maxframe/serialization/arrow.py,sha256=OMeDjLcPgagqzokG7g3Vhwm6Xw1j-Kph1V2QsIwi6dw,3513
-maxframe/serialization/core.cp38-win_amd64.pyd,sha256=j_ASMBKKkzSryNVhP4SmEqsp-WPE5tEHiZPVOr2E4fw,397824
+maxframe/serialization/core.cp38-win_amd64.pyd,sha256=pm3uyk95Wnhn8B_LPdI9pV_gHACS9uBywUES38pY2OI,397824
 maxframe/serialization/core.pxd,sha256=Fymih3Wo-CrOY27_o_DRINdbRGR7mgiT-XCaXCXafxM,1347
 maxframe/serialization/core.pyx,sha256=Qmipu3LiJGIBVy_7d4tSJqcYWnG5xj2I7IaPv2PSq5E,35078
-maxframe/serialization/exception.py,sha256=D3KhQqQLbNC_eoG_9A6j3knCAtf6I9pzJ68MPCD2ats,3073
+maxframe/serialization/exception.py,sha256=e7bZyPlZ8XhSCdeOwlYreq0HazPXKOgOA6r9Q4Ecn2Y,3113
 maxframe/serialization/maxframe_objects.py,sha256=ZHyvxIALoPuB_y3CRc70hZXyfdj4IhaKMXUEYLXHQag,1404
 maxframe/serialization/numpy.py,sha256=ENrFKl24mtYyO1vZRLwHvMD0r4z_UI7J2-yNlmfWSdk,3304
 maxframe/serialization/pandas.py,sha256=3aPzDOg9UYSI9GFpWm2aJc8EAi-d-timM8vQ8kTL3Cg,7349
 maxframe/serialization/scipy.py,sha256=fGwQ5ZreymrMT8g7TneATfFdKFF7YPNZQqgWgMa3J8M,2498
 maxframe/serialization/serializables/__init__.py,sha256=rlQhIaSAVzz4KYkc5shEHFZDPd6WDMPkxalU76yjJ3M,1406
-maxframe/serialization/serializables/core.py,sha256=9QlrQIRWSf0Ocs3YB4_n62v03RAkrb1rGN8AFCuzumI,9053
-maxframe/serialization/serializables/field.py,sha256=Rv9qFKCE0G9TSJXo1VU8mYu_OLakFsiZHC0enDrKIBs,16540
+maxframe/serialization/serializables/core.py,sha256=QUjHQPG_Qd5yh_bW-mCdapY5uKUl-s1axrM2N3eomyQ,9227
+maxframe/serialization/serializables/field.py,sha256=DVott3HAbne4UvN-heSFS9gSl0wCxV5RssS738FCjzk,16639
 maxframe/serialization/serializables/field_type.py,sha256=hkxrXT2SL_tATuobtJDfL4DzzVP2hJjDlC3PrJg6ZKo,15454
 maxframe/serialization/serializables/tests/__init__.py,sha256=29eM5D4knhYwe3TF42naTuC5b4Ym3VeH4rK8KpdLWNY,609
 maxframe/serialization/serializables/tests/test_field_type.py,sha256=uG87-bdG8xGmjrubEHCww1ZKmRupSvnNKnZoV2SnwYM,4502
@@ -363,7 +365,7 @@ maxframe/serialization/tests/__init__.py,sha256=29eM5D4knhYwe3TF42naTuC5b4Ym3VeH
 maxframe/serialization/tests/test_serial.py,sha256=9G2CbPBHINwcZ038pRwZON_OtH-JVXZ8w66BLYWP578,12923
 maxframe/tensor/__init__.py,sha256=aomZCK-bt5OYyRLGcbACxtFLrWIp14F4R3P79zwbN5E,3694
 maxframe/tensor/array_utils.py,sha256=xr_Ng-4dETJFjsMfWi5gbTPM9mRmPvRWj8QY2WKjmCg,5129
-maxframe/tensor/core.py,sha256=u9HPbYj1gOF4m7WJpjSCprIHR-3HEEwK_LhjPNmro7g,22644
+maxframe/tensor/core.py,sha256=-G-UzY81GTKj2SD9FQLqBg-UDod5LjjrEA-uF16ofms,22638
 maxframe/tensor/operators.py,sha256=8VsSZ8OcImGkSRQvrYlV05KMHGsroAYmW1o9RM2yV1U,3584
 maxframe/tensor/utils.py,sha256=An35s6MrbltYvN8WYzjKCjyozTDbGQrvUW_qz8KnA94,23632
 maxframe/tensor/arithmetic/__init__.py,sha256=SUlcG0Mf9ddgxAdydenuJ9eY5yVu0TgKfpBujI3OX4w,9695
@@ -601,26 +603,27 @@ maxframe/tensor/statistics/quantile.py,sha256=HrPxQoRXTEGf-5m79osqhUoSTFpWmIwmhO
 maxframe/tensor/ufunc/__init__.py,sha256=8QUi-cPvvbsD7i7LOeZ9sc0v1XXd7lt-XV5pQKbVZJs,821
 maxframe/tensor/ufunc/ufunc.py,sha256=XRtGlhdrW7H--mrc8fTBOlUP0mzKpd9tdRtCuLDymtc,7383
 maxframe/tests/__init__.py,sha256=29eM5D4knhYwe3TF42naTuC5b4Ym3VeH4rK8KpdLWNY,609
-maxframe/tests/test_protocol.py,sha256=n7LSC42JtGQuCRt0XY8sd9ewf6i_1C0LE3ux9yNMGp8,4993
+maxframe/tests/test_codegen.py,sha256=h3TKqP4zghxTn1twH7gR9jOe6NKXdCC1B4u0chlUrpY,2277
+maxframe/tests/test_protocol.py,sha256=IgZT2CBH1dv6V1DwSq-PjvrUhvtOf8Mab6dnWhAT3No,5331
 maxframe/tests/test_utils.py,sha256=0Iey3O6zrGI1yQU2OSpWavJNvhUjrmdkct4-27tkGUM,12353
-maxframe/tests/utils.py,sha256=ulCRwCjaJeoMo4NwvekJq5BhEJzzr9sdTn-cOx9HG_A,4697
+maxframe/tests/utils.py,sha256=gCre-8BApU4-AEun9WShm4Ff5a9a_oKxvLNneESXBjU,4732
 maxframe_client/__init__.py,sha256=xqlN69LjvAp2bNCaT9d82U9AF5WKi_c4UOheEW1wV9E,741
 maxframe_client/conftest.py,sha256=UWWMYjmohHL13hLl4adb0gZPLRdBVOYVvsFo6VZruI0,658
 maxframe_client/fetcher.py,sha256=ajt14PYVEXKShXSx-qTL_8adMCXBEsZoOhHr3yx95-M,7015
 maxframe_client/clients/__init__.py,sha256=29eM5D4knhYwe3TF42naTuC5b4Ym3VeH4rK8KpdLWNY,609
-maxframe_client/clients/framedriver.py,sha256=_Hv_pcV0Rmc3v8sNLzrZyTYz5AyoavDPOs8jtotFQWU,4570
+maxframe_client/clients/framedriver.py,sha256=upN6C1eZrCpLTsS6fihWOMy392psWfo0bw2XgSLI_Yg,4581
 maxframe_client/clients/spe.py,sha256=uizNBejhU_FrMhsgsFgDnq7gL7Cxk803LeLYmr3nmxs,3697
 maxframe_client/session/__init__.py,sha256=9zFCd3zkSADESAFc4SPoQ2nkvRwsIhhpNNO2TtSaWbU,854
 maxframe_client/session/consts.py,sha256=nD-D0zHXumbQI8w3aUyltJS59K5ftipf3xCtHNLmtc8,1380
 maxframe_client/session/graph.py,sha256=GSZaJ-PV4DK8bTcNtoSoY5kDTyyIRAKleh4tOCSUbsI,4470
 maxframe_client/session/odps.py,sha256=RG7_28UaS_8tgJUOa4ohw2QtwX2fF4yqsGGy2MksQWI,16700
-maxframe_client/session/task.py,sha256=5g3xmYpc5tY_rJqPQLok3WN-FJXDi9DarkY1b533I9s,10359
+maxframe_client/session/task.py,sha256=R8x8OERIb673vTq-o0ig6Zy2NT4_jvi8AbLhyMaljo8,11409
 maxframe_client/session/tests/__init__.py,sha256=29eM5D4knhYwe3TF42naTuC5b4Ym3VeH4rK8KpdLWNY,609
-maxframe_client/session/tests/test_task.py,sha256=lm2q1tWZiaYeU0D11_f_t7acZLLi1Ts-nlIFpZHsm4U,2684
+maxframe_client/session/tests/test_task.py,sha256=861usEURVXeTUzfJYZmBfwsHfZFexG23mMtT5IJOOm4,3364
 maxframe_client/tests/__init__.py,sha256=29eM5D4knhYwe3TF42naTuC5b4Ym3VeH4rK8KpdLWNY,609
 maxframe_client/tests/test_fetcher.py,sha256=7iYXLMIoCJLfgUkjB2HBkV-sqQ-xGlhtzfp9hRJz_kM,3605
-maxframe_client/tests/test_session.py,sha256=vTycMT6IMtPQcpVFCXFWLniHXj6m03JsSciJ7x_kAbU,6443
-maxframe-0.1.0b1.dist-info/METADATA,sha256=6EWTzzw1n2LdaYmzAdRlAN_afurzh-lLiYbljnug3Do,5146
-maxframe-0.1.0b1.dist-info/WHEEL,sha256=3SeyPJ5-Us2Ct5GSftUVKtLSlm-bNefW4m5qd0GLzww,100
-maxframe-0.1.0b1.dist-info/top_level.txt,sha256=64x-fc2q59c_vXwNUkehyjF1vb8JWqFSdYmUqIFqoTM,31
-maxframe-0.1.0b1.dist-info/RECORD,,
+maxframe_client/tests/test_session.py,sha256=s8pxf0I6PjOd6ZZQ4IYjfOM_3F3wf6SGPRMz0tAZFmo,6514
+maxframe-0.1.0b3.dist-info/METADATA,sha256=aZIVKoKmfRSjIExk2K16jB0fJj5iApw9FhoHqv6hs_k,3147
+maxframe-0.1.0b3.dist-info/WHEEL,sha256=Wb4yjwIXVKEpht4JWFUZNCzpG7JLBNZnqtK2YNdqLkI,100
+maxframe-0.1.0b3.dist-info/top_level.txt,sha256=64x-fc2q59c_vXwNUkehyjF1vb8JWqFSdYmUqIFqoTM,31
+maxframe-0.1.0b3.dist-info/RECORD,,

{maxframe-0.1.0b1.dist-info → maxframe-0.1.0b3.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.42.0)
+Generator: bdist_wheel (0.43.0)
 Root-Is-Purelib: false
 Tag: cp38-cp38-win_amd64
 

maxframe_client/clients/framedriver.py

@@ -14,6 +14,7 @@
 
 from typing import Any, Dict, List
 
+import msgpack
 from tornado import httpclient
 
 from maxframe.core import TileableGraph
@@ -28,7 +29,6 @@ from maxframe.protocol import (
 )
 from maxframe.typing_ import TimeoutType
 from maxframe.utils import (
-    deserialize_serializable,
     format_timeout_params,
     serialize_serializable,
     wait_http_response,
@@ -47,12 +47,12 @@ class FrameDriverClient:
         resp = await httpclient.AsyncHTTPClient().fetch(
             req_url, method="POST", body=serialize_serializable(req_body)
         )
-        return deserialize_serializable(resp.body).body
+        return SessionInfo.from_json(msgpack.loads(resp.body))
 
     async def get_session(self, session_id: str) -> SessionInfo:
         req_url = f"{self._endpoint}/api/sessions/{session_id}"
         resp = await httpclient.AsyncHTTPClient().fetch(req_url, method="GET")
-        return deserialize_serializable(resp.body).body
+        return SessionInfo.from_json(msgpack.loads(resp.body))
 
     async def delete_session(self, session_id: str):
         req_url = f"{self._endpoint}/api/sessions/{session_id}"
@@ -71,12 +71,12 @@ class FrameDriverClient:
             method="POST",
             body=serialize_serializable(ProtocolBody(body=req_body)),
         )
-        return deserialize_serializable(resp.body).body
+        return DagInfo.from_json(msgpack.loads(resp.body))
 
     async def get_dag_info(self, session_id: str, dag_id: str) -> DagInfo:
         req_url = f"{self._endpoint}/api/sessions/{session_id}/dags/{dag_id}"
         resp = await httpclient.AsyncHTTPClient().fetch(req_url, method="GET")
-        return deserialize_serializable(resp.body).body
+        return DagInfo.from_json(msgpack.loads(resp.body))
 
     async def wait_dag(self, session_id: str, dag_id: str, timeout: TimeoutType = None):
         query_part = format_timeout_params(timeout)
@@ -87,7 +87,7 @@ class FrameDriverClient:
             resp = await wait_http_response(
                 req_url, method="GET", request_timeout=timeout
             )
-            info = deserialize_serializable(resp.body).body
+            info = DagInfo.from_json(msgpack.loads(resp.body))
         except TimeoutError:
             info = await self.get_dag_info(session_id, dag_id)
         return info
@@ -103,7 +103,7 @@ class FrameDriverClient:
             resp = await wait_http_response(
                 req_url, method="DELETE", request_timeout=timeout
             )
-            info = deserialize_serializable(resp.body).body
+            info = DagInfo.from_json(msgpack.loads(resp.body))
         except TimeoutError:
             info = await self.get_dag_info(session_id, dag_id)
         return info

maxframe_client/session/task.py

@@ -22,6 +22,7 @@ import msgpack
 from odps import ODPS
 from odps import options as odps_options
 from odps import serializers
+from odps.errors import parse_instance_error
 from odps.models import Instance, Task
 
 from maxframe.config import options
@@ -29,6 +30,11 @@ from maxframe.core import TileableGraph
 from maxframe.protocol import DagInfo, JsonSerializable, ResultInfo, SessionInfo
 from maxframe.utils import deserialize_serializable, serialize_serializable, to_str
 
+try:
+    from maxframe import __version__ as mf_version
+except ImportError:
+    mf_version = None
+
 from .consts import (
     MAXFRAME_DEFAULT_PROTOCOL,
     MAXFRAME_OUTPUT_JSON_FORMAT,
@@ -91,6 +97,8 @@ class MaxFrameTask(Task):
             "odps.maxframe.output_format": self._output_format,
             "odps.service.endpoint": self._service_endpoint,
         }
+        if mf_version:
+            mf_opts["odps.maxframe.client_version"] = mf_version
         settings.update(mf_opts)
         self.properties["settings"] = json.dumps(settings)
         return super().serial()
@@ -104,13 +112,19 @@ class MaxFrameInstanceCaller(MaxFrameServiceCaller):
         odps_entry: ODPS,
         task_name: Optional[str] = None,
         project: Optional[str] = None,
-        priority: Optional[str] = None,
+        priority: Optional[int] = None,
         running_cluster: Optional[str] = None,
         nested_instance_id: Optional[str] = None,
         major_version: Optional[str] = None,
         output_format: Optional[str] = None,
         **kwargs,
     ):
+        if callable(odps_options.get_priority):
+            default_priority = odps_options.get_priority(odps_entry)
+        else:
+            default_priority = odps_options.priority
+        priority = priority if priority is not None else default_priority
+
         self._odps_entry = odps_entry
         self._task_name = task_name
         self._project = project
@@ -118,6 +132,7 @@ class MaxFrameInstanceCaller(MaxFrameServiceCaller):
         self._running_cluster = running_cluster
         self._major_version = major_version
         self._output_format = output_format or MAXFRAME_OUTPUT_MSGPACK_FORMAT
+
         if nested_instance_id is None:
             self._nested = False
             self._instance = None
@@ -126,9 +141,12 @@ class MaxFrameInstanceCaller(MaxFrameServiceCaller):
             self._instance = odps_entry.get_instance(nested_instance_id)
 
     def _deserial_task_info_result(
-        self, content: Union[bytes, str], target_cls: Type[JsonSerializable]
+        self, content: Union[bytes, str, dict], target_cls: Type[JsonSerializable]
     ):
-        json_data = json.loads(to_str(content))
+        if isinstance(content, (str, bytes)):
+            json_data = json.loads(to_str(content))
+        else:
+            json_data = content
         result_data = base64.b64decode(json_data["result"])
         if self._output_format == MAXFRAME_OUTPUT_MAXFRAME_FORMAT:
             return deserialize_serializable(result_data)
@@ -166,12 +184,22 @@ class MaxFrameInstanceCaller(MaxFrameServiceCaller):
             )
             return self._deserial_task_info_result(result, SessionInfo)
 
+    def _parse_instance_result_error(self):
+        result_data = self._instance.get_task_result(self._task_name)
+        try:
+            info = self._deserial_task_info_result({"result": result_data}, SessionInfo)
+        except:
+            raise parse_instance_error(result_data)
+        info.error_info.reraise()
+
     def _wait_instance_task_ready(
         self, interval: float = 0.1, max_interval: float = 5.0, timeout: int = None
     ):
         check_time = time.time()
         timeout = timeout or options.client.task_start_timeout
         while True:
+            if self._instance.is_terminated(retry=True):
+                self._parse_instance_result_error()
             status_json = json.loads(
                 self._instance.get_task_info(self._task_name, "status") or "{}"
             )

maxframe_client/session/tests/test_task.py

@@ -18,6 +18,7 @@ import os
 import mock
 from defusedxml import ElementTree
 from odps import ODPS
+from odps import options as odps_options
 
 from ...session.consts import MAXFRAME_OUTPUT_JSON_FORMAT
 from ...session.task import MaxFrameInstanceCaller, MaxFrameTask
@@ -27,17 +28,20 @@ expected_file_dir = os.path.join(os.path.dirname(__file__), "expected-data")
 
 def test_maxframe_instance_caller_creating_session():
     o = ODPS.from_environments()
-    task_caller = MaxFrameInstanceCaller(
-        odps_entry=o,
-        task_name="task_test",
-        major_version="test_version",
-        output_format=MAXFRAME_OUTPUT_JSON_FORMAT,
-        priority="100",
-        running_cluster="test_cluster",
-    )
+
+    def create_caller(**kwargs):
+        kw = dict(
+            odps_entry=o,
+            task_name="task_test",
+            major_version="test_version",
+            output_format=MAXFRAME_OUTPUT_JSON_FORMAT,
+            running_cluster="test_cluster",
+        )
+        kw.update(**kwargs)
+        return MaxFrameInstanceCaller(**kw)
 
     def mock_create(self, task: MaxFrameTask, priority=None, running_cluster=None):
-        assert priority == "100"
+        assert priority == 100
         assert running_cluster == "test_cluster"
         root = ElementTree.parse(
             os.path.join(expected_file_dir, "create_session.xml")
@@ -62,6 +66,20 @@ def test_maxframe_instance_caller_creating_session():
         target="maxframe_client.session.task.MaxFrameInstanceCaller",
         _wait_instance_task_ready=mock.DEFAULT,
         get_session=mock.DEFAULT,
-    ):
-        with mock.patch("odps.models.instances.BaseInstances.create", mock_create):
+    ), mock.patch("odps.models.instances.BaseInstances.create", mock_create):
+        task_caller = create_caller(priority=100)
+        task_caller.create_session()
+
+        old_priority = odps_options.priority
+        old_get_priority = odps_options.get_priority
+        try:
+            task_caller = create_caller(priority=100)
+            odps_options.priority = 100
+            task_caller.create_session()
+
+            odps_options.priority = None
+            odps_options.get_priority = lambda _: 100
             task_caller.create_session()
+        finally:
+            odps_options.priority = old_priority
+            odps_options.get_priority = old_get_priority

maxframe_client/tests/test_session.py

@@ -24,6 +24,7 @@ from odps import ODPS
 import maxframe.dataframe as md
 import maxframe.remote as mr
 from maxframe.core import ExecutableTuple, TileableGraph
+from maxframe.lib.aio import stop_isolation
 from maxframe.protocol import ResultInfo
 from maxframe.serialization import RemoteException
 from maxframe.session import new_session
@@ -52,6 +53,7 @@ def start_mock_session(framedriver_app):  # noqa: F811
             time.sleep(5)  # Wait for temp table deleted
         else:
             session.reset_default()
+        stop_isolation()
 
 
 def test_simple_run_dataframe(start_mock_session):