metaflow 2.13.8__tar.gz → 2.13.9__tar.gz

{metaflow-2.13.8/metaflow.egg-info → metaflow-2.13.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: metaflow
-Version: 2.13.8
+Version: 2.13.9
 Summary: Metaflow: More Data Science, Less Engineering
 Author: Metaflow Developers
 Author-email: help@metaflow.org
@@ -26,7 +26,7 @@ License-File: LICENSE
 Requires-Dist: requests
 Requires-Dist: boto3
 Provides-Extra: stubs
-Requires-Dist: metaflow-stubs==2.13.8; extra == "stubs"
+Requires-Dist: metaflow-stubs==2.13.9; extra == "stubs"
 Dynamic: author
 Dynamic: author-email
 Dynamic: classifier

{metaflow-2.13.8 → metaflow-2.13.9}/metaflow/plugins/pypi/micromamba.py

@@ -10,6 +10,7 @@ from metaflow.exception import MetaflowException
 from metaflow.util import which
 
 from .utils import MICROMAMBA_MIRROR_URL, MICROMAMBA_URL, conda_platform
+from threading import Lock
 
 
 class MicromambaException(MetaflowException):
@@ -54,22 +55,33 @@ class Micromamba(object):
             or which(os.path.join(self._path_to_hidden_micromamba, "bin/micromamba"))
         )
 
+        # We keep a mutex as environments are resolved in parallel,
+        # which causes a race condition in case micromamba needs to be installed first.
+        self.install_mutex = Lock()
+
     @property
-    def bin(self):
+    def bin(self) -> str:
         "Defer installing Micromamba until when the binary path is actually requested"
         if self._bin is not None:
             return self._bin
-        # Install Micromamba on the fly.
-        # TODO: Make this optional at some point.
-        _install_micromamba(self._path_to_hidden_micromamba)
-        self._bin = which(
-            os.path.join(self._path_to_hidden_micromamba, "bin/micromamba")
-        )
+        with self.install_mutex:
+            # another check as micromamba might have been installed when the mutex is released.
+            if self._bin is not None:
+                return self._bin
+
+            # Install Micromamba on the fly.
+            # TODO: Make this optional at some point.
+            _install_micromamba(self._path_to_hidden_micromamba)
+            self._bin = which(
+                os.path.join(self._path_to_hidden_micromamba, "bin/micromamba")
+            )
+
+            if self._bin is None:
+                msg = "No installation for *Micromamba* found.\n"
+                msg += "Visit https://mamba.readthedocs.io/en/latest/micromamba-installation.html for installation instructions."
+                raise MetaflowException(msg)
 
-        if self._bin is None:
-            msg = "No installation for *Micromamba* found.\n"
-            msg += "Visit https://mamba.readthedocs.io/en/latest/micromamba-installation.html for installation instructions."
-            raise MetaflowException(msg)
+        return self._bin
 
     def solve(self, id_, packages, python, platform):
         # Performance enhancements

metaflow-2.13.9/metaflow/version.py

@@ -0,0 +1 @@
+metaflow_version = "2.13.9"

{metaflow-2.13.8 → metaflow-2.13.9/metaflow.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: metaflow
-Version: 2.13.8
+Version: 2.13.9
 Summary: Metaflow: More Data Science, Less Engineering
 Author: Metaflow Developers
 Author-email: help@metaflow.org
@@ -26,7 +26,7 @@ License-File: LICENSE
 Requires-Dist: requests
 Requires-Dist: boto3
 Provides-Extra: stubs
-Requires-Dist: metaflow-stubs==2.13.8; extra == "stubs"
+Requires-Dist: metaflow-stubs==2.13.9; extra == "stubs"
 Dynamic: author
 Dynamic: author-email
 Dynamic: classifier

{metaflow-2.13.8 → metaflow-2.13.9}/metaflow.egg-info/requires.txt

@@ -2,4 +2,4 @@ requests
 boto3
 
 [stubs]
-metaflow-stubs==2.13.8
+metaflow-stubs==2.13.9

metaflow-2.13.8/metaflow/version.py

@@ -1 +0,0 @@
-metaflow_version = "2.13.8"