Function disable_torch_proxy

python/paddle/compat/proxy.py:503–526

Disable the PyTorch proxy by removing the TorchProxyMetaFinder from sys.meta_path. This prevents 'torch' imports from being proxied to PaddlePaddle. Example: .. code-block:: pycon >>> import paddle >>> paddle.enable_compat() # Enable torch compat globa

()

def disable_torch_proxy() -> None:
    """
    Disable the PyTorch proxy by removing the TorchProxyMetaFinder from sys.meta_path.
    This prevents 'torch' imports from being proxied to PaddlePaddle.

    Example:
        .. code-block:: pycon

            >>> import paddle
            >>> paddle.enable_compat()  # Enable torch compat globally
            >>> import torch  # type: ignore[import-not-found] # This will import paddle as torch
            >>> assert torch.sin is paddle.sin
            >>> paddle.disable_compat()  # Disable torch compat
            >>> try:
            ...     import torch  # This will raise ModuleNotFoundError
            ... except ModuleNotFoundError:
            ...     print("PyTorch compat is disabled.")
    """
    if TORCH_PROXY_FINDER in sys.meta_path:
        sys.meta_path.remove(TORCH_PROXY_FINDER)
        _clear_torch_proxy_modules()
        _copy_torch_modules_from_cache()
        return
    warnings.warn("torch compat is not installed.")


@contextmanager