diff --git a/packaging/build_conda.py b/packaging/build_conda.py index 42251812df..a131a19fc8 100644 --- a/packaging/build_conda.py +++ b/packaging/build_conda.py @@ -98,7 +98,9 @@ def _set_env_for_build(self): os.environ["TORCH_CUDA_ARCH_LIST"] = "5.0+PTX 6.0 6.1 7.0 7.5 8.0 8.6" os.environ["BUILD_VERSION"] = self._get_build_version() - tag = subprocess.check_output(["git", "rev-parse", "--short", "HEAD"], text=True).strip() + tag = subprocess.check_output( + ["git", "rev-parse", "--short", "HEAD"], text=True + ).strip() os.environ["GIT_TAG"] = tag os.environ["PYTORCH_VERSION"] = self.pytorch_version os.environ["CU_VERSION"] = self.cuda_version