config: no conda

agimus-project · Jul 20, 2023 · 3648b5e · 3648b5e
1 parent 5bdd7be
commit 3648b5e
Showing 1 changed file with 8 additions and 7 deletions.
diff --git a/happypose/pose_estimators/cosypose/cosypose/config.py b/happypose/pose_estimators/cosypose/cosypose/config.py
@@ -43,13 +43,14 @@
 MEMORY = Memory(CACHE_DIR, verbose=2)
 
 
-CONDA_PREFIX = os.environ['CONDA_PREFIX']
-if 'CONDA_PREFIX_1' in os.environ:
-    CONDA_BASE_DIR = os.environ['CONDA_PREFIX_1']
-    CONDA_ENV = os.environ['CONDA_DEFAULT_ENV']
-else:
-    CONDA_BASE_DIR = os.environ['CONDA_PREFIX']
-    CONDA_ENV = 'base'
+if 'CONDA_PREFIX' in os.environ:
+    CONDA_PREFIX = os.environ['CONDA_PREFIX']
+    if 'CONDA_PREFIX_1' in os.environ:
+        CONDA_BASE_DIR = os.environ['CONDA_PREFIX_1']
+        CONDA_ENV = os.environ['CONDA_DEFAULT_ENV']
+    else:
+        CONDA_BASE_DIR = os.environ['CONDA_PREFIX']
+        CONDA_ENV = 'base'
 
 cfg = yaml.load((PROJECT_DIR / 'happypose/pose_estimators/cosypose/config_yann.yaml').read_text(), Loader=yaml.FullLoader)