Method current

python/ray/data/context.py:994–999

(context: "DataContext")

    @staticmethod
    @contextlib.contextmanager
    def current(context: "DataContext"):
        prev: Optional[DataContext] = DataContext._set_current(context)
        try:
            yield
        finally:
            DataContext._set_current(prev)

    @staticmethod
    def _set_current(context: Optional["DataContext"]) -> Optional["DataContext"]: