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: