config

python/xorbits/__init__.py

@@ -16,7 +16,7 @@
 from . import _version
 from .core import run
 from .deploy import init, shutdown
-
+from .config import option_context, options
 
 def _install():
     from .datasets import _install as _install_datasets
@@ -44,4 +44,4 @@ def _install():
 
 __version__ = _version.get_versions()["version"]
 
-__all__ = ["init", "shutdown", "run"]
+__all__ = ["init", "shutdown", "run", "options", "option_context"]

python/xorbits/config.py

@@ -0,0 +1 @@
+from ._mars.config import option_context, options  # noqa: F401

python/xorbits/pandas/_config/config.py

@@ -66,18 +66,26 @@ def __init__(self, *args):
             except:
                 mars_dict[key] = value
 
-        self.option_contexts = mars_option_context(mars_dict)
-        self.pandas_option_context = pd.option_context(
-            *tuple(item for sublist in pd_dict.items() for item in sublist)
-        )
+        self.option_contexts = None
+        self.pandas_option_context = None
+        if mars_dict:
+            self.option_contexts = mars_option_context(mars_dict)
+        if pd_dict:
+            self.pandas_option_context = pd.option_context(
+                *tuple(item for sublist in pd_dict.items() for item in sublist)
+            )
 
     def __enter__(self):
-        self.option_contexts.__enter__()
-        self.pandas_option_context.__enter__()
+        if self.option_contexts:
+            self.option_contexts.__enter__()
+        if self.pandas_option_context:
+            self.pandas_option_context.__enter__()
 
     def __exit__(self, type, value, traceback):
-        self.option_contexts.__exit__(type, value, traceback)
-        self.pandas_option_context.__exit__(type, value, traceback)
+        if self.option_contexts:
+            self.option_contexts.__exit__(type, value, traceback)
+        if self.pandas_option_context:
+            self.pandas_option_context.__exit__(type, value, traceback)
 
 
 def set_eng_float_format(accuracy: int = 3, use_eng_prefix: bool = False) -> None: