diff --git a/setup.py b/setup.py index 14920e100da854b75ccf41789721b6f3dbaef57d..57f2e770093b50aff8f1c188721db42a77fc10ad 100644 --- a/setup.py +++ b/setup.py @@ -13,6 +13,8 @@ import time from pathlib import Path from typing import Union + + import distutils.ccompiler import distutils.command.clean from sysconfig import get_paths @@ -28,6 +30,9 @@ from wheel.bdist_wheel import bdist_wheel # Disable autoloading before running 'import torch' to avoid circular dependencies os.environ["TORCH_DEVICE_BACKEND_AUTOLOAD"] = "0" + + + from codegen.utils import PathManager BASE_DIR = os.path.dirname(os.path.realpath(__file__))