From 8fd184e393638e3a056cef8064e6e1c85bb8d158 Mon Sep 17 00:00:00 2001 From: danthe3rd Date: Thu, 12 Jan 2023 16:46:34 +0000 Subject: [PATCH] Black autoformat packaging/build_conda.py ghstack-source-id: 8713fa57c024d02e8c3fa7e5390b906eeb36295b Pull Request resolved: https://github.com/fairinternal/xformers/pull/421 __original_commit__ = fairinternal/xformers@30bb37cc95931266cc64826bf0f732a0c7a7be27 --- packaging/build_conda.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) 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