change writable location for local micromamba (#1583)

Netflix · Oct 11, 2023 · 5f373fe · 5f373fe
1 parent 545e0cb
commit 5f373fe
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/metaflow/plugins/pypi/micromamba.py b/metaflow/plugins/pypi/micromamba.py
@@ -24,8 +24,13 @@ def __init__(self):
         # micromamba is a tiny version of the mamba package manager and comes with
         # metaflow specific performance enhancements.
 
+        # METAFLOW_HOME might not be writable but METAFLOW_TOKEN_HOME might be.
+        if os.environ.get("METAFLOW_TOKEN_HOME"):
+            _home = os.environ.get("METAFLOW_TOKEN_HOME")
+        else:
+            _home = os.environ.get("METAFLOW_HOME", "~/.metaflowconfig")
         _path_to_hidden_micromamba = os.path.join(
-            os.path.expanduser(os.environ.get("METAFLOW_HOME", "~/.metaflowconfig")),
+            os.path.expanduser(_home),
             "micromamba",
         )
         self.bin = (