depyf/depyf/__init__.py at master · thuml/depyf

Repository navigation

from types import CodeType

import warnings

from .decompiler import Decompiler, decompile

try:

import torch

torch_version = torch.__version__

valid = ("dev" not in torch_version and torch_version >= "2.2") or (

"dev" in torch_version and torch_version.split("dev")[-1] >= "20231020")

if not valid:

warnings.warn(

("Please use the nightly version of PyTorch to enable bytecode hooks.\n"

"PyTorch nigh 37FE tly can be installed by: `conda install pytorch-nightly::pytorch torchvision torchaudio -c pytorch-nightly`"))

from depyf.explain.enhance_logging import install, uninstall

from depyf.explain.enable_debugging import prepare_debug, debug

except ImportError as e:

# print(e)

pass

import os

__version__ = open(f"{os.path.dirname(__file__)}/VERSION.txt").read().strip()