ob-metaflow-extensions 1.1.41__tar.gz → 1.1.42rc0__tar.gz

{ob-metaflow-extensions-1.1.41 → ob-metaflow-extensions-1.1.42rc0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ob-metaflow-extensions
-Version: 1.1.41
+Version: 1.1.42rc0
 Summary: Outerbounds Platform Extensions for Metaflow
 Author: Outerbounds, Inc.
 License: Commercial

ob-metaflow-extensions-1.1.42rc0/metaflow_extensions/outerbounds/__init__.py

@@ -0,0 +1,5 @@
+from metaflow_extensions.outerbounds.plugins.perimeters import (
+    override_metaflow_profile_with_perimeter,
+)
+
+override_metaflow_profile_with_perimeter()

ob-metaflow-extensions-1.1.42rc0/metaflow_extensions/outerbounds/plugins/perimeters.py

@@ -0,0 +1,27 @@
+import os
+import fcntl
+from os import path
+import json
+
+
+def override_metaflow_profile_with_perimeter():
+    # If OBP_CONFIG_DIR is set, use that, otherwise use METAFLOW_HOME
+    # If neither are set, use ~/.metaflowconfig
+    obp_config_dir = path.expanduser(
+        os.environ.get(
+            "OBP_CONFIG_DIR", os.environ.get("METAFLOW_HOME", "~/.metaflowconfig")
+        )
+    )
+
+    file_path = os.path.join(obp_config_dir, "ob_config.json")
+
+    if os.path.exists(file_path):
+        # Acquire a shared read lock on the file
+        fd = os.open(file_path, os.O_RDONLY)
+        fcntl.flock(fd, fcntl.LOCK_SH)
+
+        with open(file_path, "r") as f:
+            ob_config = json.loads(f.read())
+
+        if "OB_CURRENT_PERIMETER" in ob_config:
+            os.environ["METAFLOW_PROFILE"] = ob_config["OB_CURRENT_PERIMETER"]

{ob-metaflow-extensions-1.1.41 → ob-metaflow-extensions-1.1.42rc0}/ob_metaflow_extensions.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ob-metaflow-extensions
-Version: 1.1.41
+Version: 1.1.42rc0
 Summary: Outerbounds Platform Extensions for Metaflow
 Author: Outerbounds, Inc.
 License: Commercial

{ob-metaflow-extensions-1.1.41 → ob-metaflow-extensions-1.1.42rc0}/ob_metaflow_extensions.egg-info/SOURCES.txt

@@ -4,6 +4,7 @@ metaflow_extensions/outerbounds/__init__.py
 metaflow_extensions/outerbounds/config/__init__.py
 metaflow_extensions/outerbounds/plugins/__init__.py
 metaflow_extensions/outerbounds/plugins/auth_server.py
+metaflow_extensions/outerbounds/plugins/perimeters.py
 metaflow_extensions/outerbounds/plugins/kubernetes/__init__.py
 metaflow_extensions/outerbounds/plugins/kubernetes/kubernetes_client.py
 metaflow_extensions/outerbounds/toplevel/__init__.py

{ob-metaflow-extensions-1.1.41 → ob-metaflow-extensions-1.1.42rc0}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_namespace_packages
 from pathlib import Path
 
 
-version = "1.1.41"
+version = "1.1.42rc0"
 this_directory = Path(__file__).parent
 long_description = (this_directory / "README.md").read_text()