maxframe 0.1.0b1__cp39-cp39-macosx_10_9_universal2.whl → 0.1.0b3__cp39-cp39-macosx_10_9_universal2.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,40 +1,41 @@
-maxframe-0.1.0b1.dist-info/RECORD,,
-maxframe-0.1.0b1.dist-info/WHEEL,sha256=xX5eQQnems5GjABODHajeVXEXS9Ycsnxtza7cCoxqTA,113
-maxframe-0.1.0b1.dist-info/top_level.txt,sha256=64x-fc2q59c_vXwNUkehyjF1vb8JWqFSdYmUqIFqoTM,31
-maxframe-0.1.0b1.dist-info/METADATA,sha256=_D5VZblEPUpWKY4_SPzt2UAcaJWcH0ajTsB1UkpvRJM,4969
+maxframe-0.1.0b3.dist-info/RECORD,,
+maxframe-0.1.0b3.dist-info/WHEEL,sha256=Ber5qIqMO9Bgjzsdsa-quaSLD92qE_yQT1ahu4qEUHc,113
+maxframe-0.1.0b3.dist-info/top_level.txt,sha256=64x-fc2q59c_vXwNUkehyjF1vb8JWqFSdYmUqIFqoTM,31
+maxframe-0.1.0b3.dist-info/METADATA,sha256=056zhE5bxFHo_1CAOnpi1A2Ov1JI8hg435HsygF02lA,3043
 maxframe_client/conftest.py,sha256=7cwy2sFy5snEaxvtMvxfYFUnG6WtYC_9XxVrwJxOpcU,643
 maxframe_client/__init__.py,sha256=3b-z0oFVVwtIzVFBxOb9pw7gz4IhTSh4FiHtVgnxS4Q,724
 maxframe_client/fetcher.py,sha256=Ys_qu2qtniXuj9YSfeHvevdrAAEgm8k4YjyoZusdVmg,6813
 maxframe_client/clients/__init__.py,sha256=FEFOVLi3o2GpZoedTtLYvbie0eQBehJIjtCrWca2ZHw,596
 maxframe_client/clients/spe.py,sha256=ArZMNQ7olicI4O1JO7CyRP7-hb60DF71ZKCTO0N39uE,3593
-maxframe_client/clients/framedriver.py,sha256=NtG3QjZpuMmxZl9IWJRAAzWLojaWn6k8vy0YRTq6hqU,4452
-maxframe_client/tests/test_session.py,sha256=Z86Zv2p1ECpiZDNK7YitRXoEIDmhNWbRJB0HSyX462Y,6238
+maxframe_client/clients/framedriver.py,sha256=Rn09529D2qBTgNGc0oCY0l7b3FgzT87TqS1nujGQaHw,4463
+maxframe_client/tests/test_session.py,sha256=75mxU5UTWjb1loQLo9BAkg7BzKuj4vYX49j4vP1-8fA,6307
 maxframe_client/tests/__init__.py,sha256=FEFOVLi3o2GpZoedTtLYvbie0eQBehJIjtCrWca2ZHw,596
 maxframe_client/tests/test_fetcher.py,sha256=q7kYCznM6WSxx9TCbHrxs7Zy1L2a5zu9D-Pi1XNgQzg,3516
-maxframe_client/session/task.py,sha256=IjgeN1DsN918x7qEF6cZrCkekkVnM3yTVaJklDzC3No,10080
+maxframe_client/session/task.py,sha256=z5j8qtBM6cs_lZrvfy4Ji3F3sVOhPOCr5r1RsNe7rN4,11102
 maxframe_client/session/graph.py,sha256=nwILNOIVaIf4E3xWffTAAlRsKRYU_zGW3oVO10du8Xw,4351
 maxframe_client/session/__init__.py,sha256=KPqhSlAJiuUz8TC-z5o7mHDVXzLSqWwrZ33zNni7piY,832
 maxframe_client/session/consts.py,sha256=R37BxDF3kgCy0qmDdwLaH5jB7mb7SzfYV6g9yHBKAwk,1344
 maxframe_client/session/odps.py,sha256=uuPk5rc1OpFk9PNFU1R6r4HQXOHAOU7ZVvbcdGk_N_s,16248
-maxframe_client/session/tests/test_task.py,sha256=MvizDb-WXbJn-cZ_e6rozO-VT95Expxu6ngyeD0TyZA,2617
+maxframe_client/session/tests/test_task.py,sha256=lDdw3gToaM3xSaRXEmHUoAo2h0id7t4v_VvpdKxQAao,3279
 maxframe_client/session/tests/__init__.py,sha256=FEFOVLi3o2GpZoedTtLYvbie0eQBehJIjtCrWca2ZHw,596
 maxframe/_utils.pyx,sha256=I4kmfhNr-xXK2ak22dr4Vwahzn-JmTaYctbL3y9_UBQ,17017
 maxframe/conftest.py,sha256=ZzwKhGp7LAVpzQYJkniwIUQqIegcaDQAhyzDyU2qld4,4264
-maxframe/opcodes.py,sha256=iKI6CuYqVwgk6YR9ODAW6eM67HPoMRxlPEZMM980CO4,9960
+maxframe/opcodes.py,sha256=L1_GGenexNnD4dk1ueelgpAebHCC-cY7d--1QPhw8l4,9984
 maxframe/env.py,sha256=_K499f7giN7Iu9f39iI9p_naaEDoJ0rx8dInbzqFOVI,1402
 maxframe/mixin.py,sha256=HBAeWYGb7N6ZIgkA-YpkKiSY1GetcEVNTuMb0ieznBs,3524
-maxframe/protocol.py,sha256=PGzploysEiSusFZ9AMroEa02eAjHWuEFwtEoE2WDdsY,13373
+maxframe/protocol.py,sha256=LjjE6iw0ZVx82tBMbff4izkGuiJxRG0MTOaPYYpRL10,14190
 maxframe/session.py,sha256=TspAq0EUhi-7VQb1kAuQyjhpX8vMYqgdk4vzbGCkJcY,35344
 maxframe/__init__.py,sha256=YGnga-nYEppPoDZoZ5s64PZ0RYLaWtcYtmYSLTjKUBE,976
-maxframe/utils.py,sha256=JNbooQGsOdK0gDhSEDFKB7gbWolmRjh9I2rMtaUoq3A,33369
+maxframe/utils.py,sha256=BrMJLO0-iPKABkNztXWpqiQWHoe-X75HGiCJr5aeles,33914
 maxframe/extension.py,sha256=4IzF9sPyaRoAzLud0iDLooOgcyq4QunXH0ki3q9Hn8I,2374
-maxframe/udf.py,sha256=QGbNIh3L3lL-GFD_oCm1wnwSeIy61k6mba0sMGul6nE,2257
-maxframe/_utils.cpython-39-darwin.so,sha256=mX12_10230-OCyj6gC9kYPU_4vD6khvk9KEhDzzyRN8,914120
+maxframe/errors.py,sha256=nQZoLGdLJz-Uf9c2vUvQl08QMvRqtkBOhKfdPYRZA4o,690
+maxframe/udf.py,sha256=tWOMTkNqGWw3ZpNC9wEU0GGNSBV8sV7E-Ye80DI28eg,2241
+maxframe/_utils.cpython-39-darwin.so,sha256=EP-V5iInmDqKvq5RZwMYGuXUUGbLY8nnjh3f_Di6xbQ,914504
 maxframe/typing_.py,sha256=fzHETru3IOZAJwU9I7n_ib3wHuQRJ9XFVmAk7WpqkMo,1096
-maxframe/codegen.py,sha256=mqMIphJFN8r5jtiPrs51QuoBiZoOuYsHBcEeKnUBb-Y,13606
+maxframe/codegen.py,sha256=pSiGHoEo9YajFPHbFHvi7fGkiJmAQdBCe0mMXNOG6-U,15846
 maxframe/_utils.pxd,sha256=AhJ4vA_UqZqPshi5nvIZq1xgr80fhIVQ9dm5-UdkYJ8,1154
 maxframe/dataframe/arrays.py,sha256=RWzimUcrds5CsIlPausfJAkLUjcktBSSXwdXyUNKEtU,28752
-maxframe/dataframe/__init__.py,sha256=ovkGGY3hOJIHlRvbPotYQxr2HeIRQOa0bBxxUDtVNqo,2036
+maxframe/dataframe/__init__.py,sha256=tpbt4OgW4hoU_1OsQu1WSxbpZD4YPXW_oT4xOxNPNqE,2112
 maxframe/dataframe/core.py,sha256=mJ5I_qBP8HDAzlwPnkj8RN4EyBHC8d5unbe2LC6HKXg,73641
 maxframe/dataframe/initializer.py,sha256=4BpZJB8bbyFnABUYWBrk_qzzrogEsWgFuU21Ma9IsjY,10264
 maxframe/dataframe/utils.py,sha256=qWRo51rcMTlo4mvZ8ZZq1zIF9CiAgU1qRtoCAaYrR34,44111
@@ -47,7 +48,7 @@ maxframe/dataframe/statistics/tests/__init__.py,sha256=FEFOVLi3o2GpZoedTtLYvbie0
 maxframe/dataframe/misc/accessor.py,sha256=QXjV5Q6LW0KNs67X5h2bCBhwHV8R7riP6TmTY_ECpmA,10027
 maxframe/dataframe/misc/describe.py,sha256=Qvmv_5rj-fk_ABcS5h8IxNHY-EAPB1rG446LiB1MRgU,4387
 maxframe/dataframe/misc/astype.py,sha256=OkGDO6PBZli46m0kKq0BOO2OOwJWEyatzeBFV_mnKz8,7797
-maxframe/dataframe/misc/to_numeric.py,sha256=8F7pmTXBR4-ppgeoa_9swDuZ8yiqYCaDlXficXpDTwE,6196
+maxframe/dataframe/misc/to_numeric.py,sha256=gG5fxEtC71SkDeHaJ0Vd1JZrIM8gMHGFaiwyFtMwbWs,6262
 maxframe/dataframe/misc/check_monotonic.py,sha256=apHehWNRUTuzux3hI4JflWiuvpaeEFwtpPTWVide7wU,2422
 maxframe/dataframe/misc/datetimes.py,sha256=fgLczaX-yDdairnNW1EsabEzoioP5gUsO0ukYANisLE,2503
 maxframe/dataframe/misc/string_.py,sha256=eG4uMA6ScB0Pfby6pBLH24Ln01Aai2aAn-hMyRM5YFo,8101
@@ -82,13 +83,14 @@ maxframe/dataframe/datasource/from_index.py,sha256=2061zsQn-BhyHTT0X9tE0JK8vLxQU
 maxframe/dataframe/datasource/dataframe.py,sha256=LxAKF4gBIHhnJQPuaAUdIEyMAq7HTfiEeNVls5n4I4A,2023
 maxframe/dataframe/datasource/series.py,sha256=QcYiBNcR8jjH6vdO6l6H9F46KHmlBqVCTI2tv9eyZ9w,1909
 maxframe/dataframe/datasource/__init__.py,sha256=C8EKsHTJi-1jvJUKIpZtMtsK-ZID3dtxL1voXnaltTs,640
+maxframe/dataframe/datasource/read_odps_query.py,sha256=nj8l38S0iVAXgNXgzDFERO-HNp6lJ1GahImwDpEzXXw,9821
 maxframe/dataframe/datasource/core.py,sha256=ozFmDgw1og7nK9_jU-u3tLEq9pNbitN-8w8XWdbKkJ0,2687
 maxframe/dataframe/datasource/date_range.py,sha256=CDGpxDyjLwnb66j-MIiiTfXGXHGh5MLhEmj6x2riIlU,17244
-maxframe/dataframe/datasource/read_odps_table.py,sha256=2FIgD2N6IOFj9IbpwOCX_XYOmFpOugLY2Byl6-jtVsI,9018
+maxframe/dataframe/datasource/read_odps_table.py,sha256=VCqWxJswcjujVoUNXb2kbTkOZroMkFCg5n272Yn7ME4,9045
 maxframe/dataframe/datasource/read_parquet.py,sha256=9auOcy8snTxCOohgXZCUXfT_O39irdkBngZH5svgx0E,14531
 maxframe/dataframe/datasource/from_tensor.py,sha256=4viuN5SLLye7Xeb8kouOpm-osoQ2yEovWTDNPQuW8gE,14727
-maxframe/dataframe/datasource/from_records.py,sha256=Ppusfo6HZtkVgdvu8Qgm1I-Osm3Cbk50tx9tnYvzXXQ,3375
-maxframe/dataframe/datasource/tests/test_datasource.py,sha256=beWaNWumBuLxMS8kDsTlLgimibME1MNuGoWyQXeo3P8,11327
+maxframe/dataframe/datasource/from_records.py,sha256=WBYouYyg7m_8NJdN-yUWSfJlIpm6DVP3IMfLXZFugyI,3442
+maxframe/dataframe/datasource/tests/test_datasource.py,sha256=4O3N-XD-MpJxEQfILu4cS7gU82hqgS9g9gnDDEsw56k,14640
 maxframe/dataframe/datasource/tests/__init__.py,sha256=FEFOVLi3o2GpZoedTtLYvbie0eQBehJIjtCrWca2ZHw,596
 maxframe/dataframe/sort/sort_index.py,sha256=Hwbkm9x8kqGgXh4gMcAtYMDjYtt-S3CJXfYR9pN5Iqk,5412
 maxframe/dataframe/sort/__init__.py,sha256=Vt2Ynr7uAX51hLbQu93QeHiFH4D9_WJMO99KljpbO2U,1160
@@ -126,9 +128,9 @@ maxframe/dataframe/extensions/tests/__init__.py,sha256=FEFOVLi3o2GpZoedTtLYvbie0
 maxframe/dataframe/groupby/aggregation.py,sha256=BuhqZal6RLHkjvwJep86oT1a7rMqxxUAPxQ_dao6I6E,11953
 maxframe/dataframe/groupby/fill.py,sha256=JF3NxyXigZqg8ecKtLSndDmNMX8S6g_ChQR9JAK036E,4721
 maxframe/dataframe/groupby/cum.py,sha256=jdGQm7U-GosgHMfneHZC5Z2uraj6iBmSFOhqP3m18B0,3755
-maxframe/dataframe/groupby/__init__.py,sha256=010xiNLGuIREN4188o6GlFSLUGjUk5vFFC1vhyHykks,3247
+maxframe/dataframe/groupby/__init__.py,sha256=nZkz1OAdYRj8qwQkUAZDax0pfCsUH_fprwuksS97vuc,3361
 maxframe/dataframe/groupby/getitem.py,sha256=kUcI9oIrjOcAHnho96Le9yEJxFydALsWbGpZfTtF8gY,3252
-maxframe/dataframe/groupby/core.py,sha256=AHO_6tX0fL1YkXLXUm2FjhPvxnmfmj-lkx711Uk5TPM,5980
+maxframe/dataframe/groupby/core.py,sha256=K1hg9jod6z3C65SYoidmEAd_k0Mear4l5IQuwNMjpxQ,6075
 maxframe/dataframe/groupby/transform.py,sha256=pY3WPA4gN8piYSTInncjnRdh8mi9FDQa00A-Pyaoil4,8586
 maxframe/dataframe/groupby/head.py,sha256=ZDkbSn3HuUR4GGkZJqo_fL-6KFJfs55aKXQkAh_0wvA,3266
 maxframe/dataframe/groupby/sample.py,sha256=IdoyzT-V5309txYvM_iaYKupsULfozMGwm1K3oihTf4,6935
@@ -171,8 +173,8 @@ maxframe/dataframe/plotting/__init__.py,sha256=JShRu5lSUAfwgTFUMxQoIqiH1gg9H5cie
 maxframe/dataframe/plotting/core.py,sha256=-Ei1ZMQbaTSC6tk8olHmoLuxL7C7NbhKQ6gnSpBuHFw,2193
 maxframe/dataframe/plotting/tests/test_plotting.py,sha256=IKAOcG0LotMGZCrvDzWiHmoFXqtyrVDn87eIOdyg6Ho,4118
 maxframe/dataframe/plotting/tests/__init__.py,sha256=FEFOVLi3o2GpZoedTtLYvbie0eQBehJIjtCrWca2ZHw,596
-maxframe/dataframe/window/aggregation.py,sha256=DGjlCia2BSeCNejNItRVp5Njltgai0oHcGzBqxp36M4,4452
-maxframe/dataframe/window/ewm.py,sha256=wWUMYo67FEaRaFEKjB9BhDzjK2gnFQrvZxK9U16q-vg,8032
+maxframe/dataframe/window/aggregation.py,sha256=ncdJavwZcj6LWfF_W9wWUWAIjOxu7DOEEW16y5EQCF4,3794
+maxframe/dataframe/window/ewm.py,sha256=xqRiw4_KYpPTzhdKnN9wIknWdKsFls-uEzYO1T18vDk,7789
 maxframe/dataframe/window/__init__.py,sha256=D8DOhmH-PabNAFPNwzwsw-mH0iELIwyU7tyiiaZGJhw,910
 maxframe/dataframe/window/core.py,sha256=T15k5OqC1y1ewlT2ddDf0oFP9LeOdneZ3po1mxLdOgc,2205
 maxframe/dataframe/window/rolling.py,sha256=btt0uOtsK3z2gR96tVXrXJUrvLFus5FUunRaooizpIw,12042
@@ -180,7 +182,7 @@ maxframe/dataframe/window/expanding.py,sha256=OmrNK14cj9corIyTCACZy1_SgTXL3i2ZQp
 maxframe/dataframe/window/tests/test_expanding.py,sha256=RV982SmVAVuUVS7cjWdTFk_VqACphhRQb9DPu9eyWqQ,1944
 maxframe/dataframe/window/tests/test_rolling.py,sha256=JWnXGUCz3li5B10rvnzzcTXbQq1lL6Svo3NL7BeBGGU,1714
 maxframe/dataframe/window/tests/__init__.py,sha256=FEFOVLi3o2GpZoedTtLYvbie0eQBehJIjtCrWca2ZHw,596
-maxframe/dataframe/window/tests/test_ewm.py,sha256=aFLI7GDddC3n7bht643SoV6QDWaUxqxRhhqIdySEdaw,2254
+maxframe/dataframe/window/tests/test_ewm.py,sha256=R6I2smeLhpS9cczCAuZdP5nKj0s4ssKB5VJ6hRW-9b0,2060
 maxframe/dataframe/arithmetic/arctan.py,sha256=zFCdG8rCmRmU3p9jGm3tQp5ulAsx4nmKn3m8e7jY6nk,930
 maxframe/dataframe/arithmetic/arccos.py,sha256=WF3SE975VbwCOsrSPNck4swIejDFI-tVm9H4DQrBv6Q,930
 maxframe/dataframe/arithmetic/arctanh.py,sha256=7H32t46reUr0k_4pEj2Dx_dAkH59OIUinedqoOPd0Hg,935
@@ -265,15 +267,15 @@ maxframe/core/entity/tileables.py,sha256=b9jn_OQ-FQkbw7E7jMLjoJ4-VR7tBS8Mbx_j4iZ
 maxframe/core/entity/__init__.py,sha256=Hz_p6eTkrSdkT7YCo5aeGZ33tms5wwifMp4TeYsAVlw,1292
 maxframe/core/entity/core.py,sha256=t7Ex9Yb7A1h_XwyRG88Fx4ZOai-NQKi2luRVS_jFPEo,4018
 maxframe/core/entity/utils.py,sha256=IuNgFmBQFRioAA1hgZe6nTEggOmDY-iooZqncQQrV28,942
-maxframe/core/entity/executable.py,sha256=mkG6Ao6f4jAPhgGlPg4KOCwXIhdEZ5I69AEf3VF2iGQ,10903
+maxframe/core/entity/executable.py,sha256=HKXHXdPIyxg9i-OWmJxIY3KfXwX0x3xN9QcR5Xhc7dQ,10938
 maxframe/core/entity/output_types.py,sha256=uqApvFK8w6_aMxRets69dTwD1ndBDgVgqDCflyt9ubg,2645
-maxframe/core/entity/objects.py,sha256=3bhwSSzVD0o_UEzsXie-SuCie9y_mma9VDblpy3XmG8,3020
+maxframe/core/entity/objects.py,sha256=RMHLTGbIHZNxxX59lAuQydAKcR32qKleIYUqdElGS4E,3034
 maxframe/core/entity/fuse.py,sha256=47U6MHRzA2ZvUi-kJb7b3mC_gN07x3yebBgX2Jj7VZo,2277
 maxframe/core/entity/chunks.py,sha256=yNSLCWOpA_Z6aGr6ZI32dIJf3xPdRBWbvdsl8sTM3BE,2134
 maxframe/core/graph/__init__.py,sha256=rnsXwW0ouh1f7SVtq73-PzLE-MBM6Op_0l6J7b7wGRE,821
-maxframe/core/graph/core.cpython-39-darwin.so,sha256=Q872C6kh4yYEvOMUDAyOD5jr_n91gFXxmHa9wzNaoXU,731414
+maxframe/core/graph/core.cpython-39-darwin.so,sha256=QUFwgB7giYUpx3RDJWE89do1hcnUPoOvc4DzHCoHGP8,731414
 maxframe/core/graph/entity.py,sha256=56gjXyDXN-TTPm3AQOxuRVQbb_fguKFDL_Xm7i95XEk,5559
-maxframe/core/graph/core.pyx,sha256=6G5tcB6Dt8sHv6l_Xfg-BHWyq0PWZmVDAnRc7yd11Uw,15853
+maxframe/core/graph/core.pyx,sha256=ZJPx_MTOBMaX-6mns6tAiu-wrIBvRAKN44YAGTypJ1Y,15887
 maxframe/core/graph/tests/__init__.py,sha256=FEFOVLi3o2GpZoedTtLYvbie0eQBehJIjtCrWca2ZHw,596
 maxframe/core/graph/tests/test_graph.py,sha256=kZfe_SfMOoHjEfXvtVtn0RjK4lOd-tFasxSpfL4G1WE,7462
 maxframe/core/graph/builder/__init__.py,sha256=BV5tAiZ-bJE0FBGSCEnfqKNwBNHgvB9aSzEp0PY6SnA,702
@@ -289,30 +291,30 @@ maxframe/core/operator/core.py,sha256=T2YYIleKiFxmufpHF7ELOztlJsYO-Kn4KYhQhY4jRU
 maxframe/core/operator/shuffle.py,sha256=A_w62UxXRKyDNESo4Q0UGmCpuFfPlQXWmPyQGHdpz3A,4746
 maxframe/core/operator/objects.py,sha256=SJoF5GBusb1_j3-dSnfKSndr9WTrlU3Db-fCrcMs3jI,2152
 maxframe/core/operator/fuse.py,sha256=0RGemF99gQCwV4aEk-K6T5KAGToO-487dFk8LyYDIZw,925
-maxframe/core/operator/base.py,sha256=VgE_osUBvXpvfDyKMDBCDe24Gqsmd6VIp92bQITcV6U,14841
+maxframe/core/operator/base.py,sha256=nxuSKjbBzDrItM9PGmFo8RLwParazu525jMLWj0kXkM,15251
 maxframe/core/operator/tests/test_core.py,sha256=57aICnc5VLqdVK7icAORTWC81bSjBxeeVWIJcha9J_0,1691
 maxframe/core/operator/tests/__init__.py,sha256=FEFOVLi3o2GpZoedTtLYvbie0eQBehJIjtCrWca2ZHw,596
-maxframe/config/config.py,sha256=O268Knu9-rPVUQa3UmGEm7wtJbaSNpzKcRaMPzRgpOo,12758
+maxframe/config/config.py,sha256=3lQj99eMGg9MBW1gMJAdGFD88UEcdZf71sHgAHXASAk,13045
 maxframe/config/validators.py,sha256=2m9MrkjDUFiU4PPaWIw8tjwMaOy8AYmuJFqVnnY8IMY,1615
 maxframe/config/__init__.py,sha256=g5lN3nP2HTAXa6ExGxU1NwU1M9ulYPmAcsV-gU7nIW8,656
 maxframe/config/tests/__init__.py,sha256=FEFOVLi3o2GpZoedTtLYvbie0eQBehJIjtCrWca2ZHw,596
 maxframe/config/tests/test_validators.py,sha256=U_7yKSl0FdVdDwKU1EsnCzNWaOXi8xrIC08x7hb_O4c,1240
 maxframe/config/tests/test_config.py,sha256=Cx3buKli4F77zwJaB-vnUSlE9LUbt3ObHW38cIE2dDs,2736
-maxframe/serialization/exception.py,sha256=YLt2lZCWfEZr0elSYus9JhueCwG_yV3RLJkcEKj2W9E,2988
+maxframe/serialization/exception.py,sha256=TIlEDiuYwVgYkN4Se0ydXn2KzEJy4Yv8w2XwlTBqDdM,3027
 maxframe/serialization/core.pxd,sha256=51XCBozW9j_E0sBgyYkHf5xNfqCMJb36Fwm_Pxv_rZs,1303
 maxframe/serialization/pandas.py,sha256=D4_H4KjAl8DRQtODh9VBP94yB8ce1cYQUKrzXtn10KE,7147
 maxframe/serialization/arrow.py,sha256=VnGxNLU9UV_cUPTze43bEFCIbYLAOZnp2pAwVJbAIzQ,3418
 maxframe/serialization/__init__.py,sha256=9eSnoDww1uw2DAXEBBTB2atJQHzd-38XVxrCkoaypxA,921
 maxframe/serialization/maxframe_objects.py,sha256=R9WEjbHL0Kr56OGkYDU9fcGi7gII6fGlXhi6IyihTsM,1365
 maxframe/serialization/numpy.py,sha256=8_GSo45l_eNoMn4NAGEb9NLXY_9i4tf9KK4EzG0mKpA,3213
-maxframe/serialization/core.cpython-39-darwin.so,sha256=qhouqwNpE6ZbCiCJggz6CW8nQiUtbino5A9qqK4lDLk,1308182
+maxframe/serialization/core.cpython-39-darwin.so,sha256=5If5ChKRLCZcYL8cVKH-EP3angUznykEXjnsqe76_7M,1308182
 maxframe/serialization/scipy.py,sha256=hP0fAW0di9UgJrGtANB2S8hLDbFBtR8p5NDqAMt5rDI,2427
 maxframe/serialization/core.pyx,sha256=AATN47RdBTq2zg7--3xX2VHyAZSvoAuYRt7B7gEgKPE,33984
 maxframe/serialization/tests/test_serial.py,sha256=Wj_I6CBQMaOtE8WtqdUaBoU8FhBOihht6SfeHOJV-zU,12511
 maxframe/serialization/tests/__init__.py,sha256=FEFOVLi3o2GpZoedTtLYvbie0eQBehJIjtCrWca2ZHw,596
-maxframe/serialization/serializables/field.py,sha256=OjQyF667xANnNTVpQ34etd2LTNZrNUW_1K5lYuVWI1U,15918
+maxframe/serialization/serializables/field.py,sha256=atVgX-9rsVG1fTev7vjQArVwIEaCRjXoSEjpQ3mh6bA,16015
 maxframe/serialization/serializables/__init__.py,sha256=_wyFZF5QzSP32wSXlXHEPl98DN658I66WamP8XPJy0c,1351
-maxframe/serialization/serializables/core.py,sha256=a_6_UgomggS5ty5LXycnAphjFiqA0nEqgNUOQGN04pk,8796
+maxframe/serialization/serializables/core.py,sha256=xlqVUlBK3aLTavHLWHg4JXUTaBGzSuM7t-XHahB8et4,8965
 maxframe/serialization/serializables/field_type.py,sha256=Feh09hu8XyaxS5MaJ4za_pcvqJVuMkOeGxwQ9OuJw6I,14865
 maxframe/serialization/serializables/tests/test_field_type.py,sha256=T3ebXbUkKveC9Pq1nIl85e4eYascFeJ52d0REHbz5jo,4381
 maxframe/serialization/serializables/tests/__init__.py,sha256=FEFOVLi3o2GpZoedTtLYvbie0eQBehJIjtCrWca2ZHw,596
@@ -328,12 +330,13 @@ maxframe/odpsio/tests/test_schema.py,sha256=yss1ly55ErYse95XMFq2s_GWL8UnwZga5RyN
 maxframe/odpsio/tests/test_arrow.py,sha256=SQ9EmI9_VOOC8u6Rg6nh3IPC2fPbLvJ9HwtpMNDRhL8,3106
 maxframe/odpsio/tests/test_volumeio.py,sha256=UEqFANuPKyFtlIh2JNi-LoixH52bxsgHdxu3himnEvs,3022
 maxframe/tests/test_utils.py,sha256=xaAoURr5NOJUTY0XVa2H8qOStcEH5UQSXItkatHFxFE,11977
-maxframe/tests/test_protocol.py,sha256=GnA-czIKHvNqZCBbgnWQZ1CP06_SorTEvGZ9VHUGvK4,4857
+maxframe/tests/test_protocol.py,sha256=t11yxh4_gWxxCuk09zo3pn9Nn96DBBQTBt12ewKDwLQ,5187
 maxframe/tests/__init__.py,sha256=FEFOVLi3o2GpZoedTtLYvbie0eQBehJIjtCrWca2ZHw,596
-maxframe/tests/utils.py,sha256=DHpzJLs-U6xiv5jSHe0m1D2b5bk7WLA7fa2-PY4HpOg,4534
+maxframe/tests/utils.py,sha256=wJtSFXt3BD4i5zdO4JBQk_kNAxrtyGLro0jodCA4xuY,4568
+maxframe/tests/test_codegen.py,sha256=GMrnpSb2eyB_nmuv8-_p47Kw877ElKS3BP52SpqZNIQ,2208
 maxframe/lib/wrapped_pickle.py,sha256=xJa0wI-GsBZFKQpVnlh_hZBlQ2u1D8VO2aBIW7VOdP4,3810
 maxframe/lib/version.py,sha256=yQ6HkDOvU9X1rpI49auh-qku2g7gIiztgEH6v1urOrk,18321
-maxframe/lib/mmh3.cpython-39-darwin.so,sha256=nFLAK7mjgXYuEBzGbzKsQW_d3OFgv_0Cfz68t3njMEE,120374
+maxframe/lib/mmh3.cpython-39-darwin.so,sha256=dsCMn_BjSE7J_loHlEuT7bb2ND0aQWW9I6TOXtNkCvo,120374
 maxframe/lib/compression.py,sha256=k9DSrl_dNBsn5azLjBdL5B4WZ6eNvmCrdMbcF1G7JSc,1442
 maxframe/lib/__init__.py,sha256=CzfbLNqqm1yR1i6fDwCd4h1ptuKVDbURFVCb0ra7QNc,642
 maxframe/lib/functools_compat.py,sha256=PMSkct9GIbzq-aBwTnggrOLNfLh4xQnYTIFMPblzCUA,2616
@@ -344,7 +347,7 @@ maxframe/lib/tests/__init__.py,sha256=FEFOVLi3o2GpZoedTtLYvbie0eQBehJIjtCrWca2ZH
 maxframe/lib/tests/test_wrapped_pickle.py,sha256=oz1RLwHSZstXgw4caNeaD0ZgQZvkzDLsx7hFN-NvP7U,1524
 maxframe/lib/cython/__init__.py,sha256=FEFOVLi3o2GpZoedTtLYvbie0eQBehJIjtCrWca2ZHw,596
 maxframe/lib/cython/libcpp.pxd,sha256=2o9HWtyCMtN9xk7WH_mq1i89IbMPwfZA8yHETjRALXs,1100
-maxframe/lib/aio/isolation.py,sha256=p2QuFBVe1ktX8uVFbu6_U1yNzbhBNpFZ9f4A6-Y6XHo,2628
+maxframe/lib/aio/isolation.py,sha256=2nA16GdOXEUNVVdxQXbmU4YvY6wPG2oSV4z1x9uW6Gw,2761
 maxframe/lib/aio/__init__.py,sha256=1_nx7d5AqXRwa7ODzVfL8gH-tsDAH4YyY-JFPC8TA6w,936
 maxframe/lib/aio/_threads.py,sha256=tr7FYViGT7nyR7HRw3vE3W-9r3-dZ1IP_Kbhe9sgqpw,1328
 maxframe/lib/aio/file.py,sha256=aZF8NkkccsVsOniWMBiPqPSk48bb7zGRPB2BegWRaqM,2012
@@ -384,7 +387,7 @@ maxframe/lib/tblib/cpython.py,sha256=FQ0f6WTQyQHoMRhgPqrA0y0Ygxlbj5IC53guxA4h9Cw
 maxframe/lib/tblib/decorators.py,sha256=bcllK3kVuPnj6SNZGmlJGxTK0ovdt7TJDXrhA4UE5sQ,1063
 maxframe/tensor/array_utils.py,sha256=259vG4SjyhiheARCZeEnfJdZjoojyrELn41oRcyAELs,4943
 maxframe/tensor/__init__.py,sha256=-kir8LUsXCDGcc7YdKqWgNEHSrgU_HE5uPam0jLLP6g,3511
-maxframe/tensor/core.py,sha256=x9HRQbMjhXJTtPQjKhkN42qZaPQIR7uvkMkOpqUspL8,21920
+maxframe/tensor/core.py,sha256=Ojxaf5b8sJ6ZZGezyFHQJ5XsSpUrBOnZgFeUQgpVJpI,21914
 maxframe/tensor/utils.py,sha256=bwVN0iuVic1tpFai6Hk-1tQLqckQ2IYS7yZKMTcOU1I,22914
 maxframe/tensor/operators.py,sha256=iGkDIRz152gXrPb5JbqOvXngpq3QaCg-aNO4gHZPLN0,3461
 maxframe/tensor/statistics/quantile.py,sha256=UFzTmBwgNL7k_QOJ84qPfycQrW8MyOa1gcp-uFsylIY,9484

{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: cp39-cp39-macosx_10_9_universal2
 

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):