ob-metaflow-extensions 1.1.70__py2.py3-none-any.whl → 1.1.71__py2.py3-none-any.whl

metaflow_extensions/outerbounds/plugins/perimeters.py

@@ -7,6 +7,9 @@ from typing import Union
 
 CURRENT_PERIMETER_KEY = "OB_CURRENT_PERIMETER"
 CURRENT_PERIMETER_URL = "OB_CURRENT_PERIMETER_MF_CONFIG_URL"
+CURRENT_PERIMETER_URL_LEGACY_KEY = (
+    "OB_CURRENT_PERIMETER_URL"  # For backwards compatibility with workstations.
+)
 
 
 def get_perimeter_config_url_if_set_in_ob_config() -> Union[str, None]:
@@ -37,10 +40,18 @@ def get_perimeter_config_url_if_set_in_ob_config() -> Union[str, None]:
         with open(file_path, "r") as f:
             ob_config = json.loads(f.read())
 
-        if CURRENT_PERIMETER_KEY in ob_config and CURRENT_PERIMETER_URL in ob_config:
+        if CURRENT_PERIMETER_KEY in ob_config and (
+            CURRENT_PERIMETER_URL in ob_config
+            or CURRENT_PERIMETER_URL_LEGACY_KEY in ob_config
+        ):
             os.environ[CURRENT_PERIMETER_KEY] = ob_config[CURRENT_PERIMETER_KEY]
-            os.environ[CURRENT_PERIMETER_URL] = ob_config[CURRENT_PERIMETER_URL]
-            return ob_config[CURRENT_PERIMETER_URL]
+            if CURRENT_PERIMETER_URL in ob_config:
+                os.environ[CURRENT_PERIMETER_URL] = ob_config[CURRENT_PERIMETER_URL]
+            elif CURRENT_PERIMETER_URL_LEGACY_KEY in ob_config:
+                os.environ[CURRENT_PERIMETER_URL] = ob_config[
+                    CURRENT_PERIMETER_URL_LEGACY_KEY
+                ]
+            return os.environ[CURRENT_PERIMETER_URL]
         else:
             raise MetaflowException(
                 "{} does not contain the key {}".format(

metaflow_extensions/outerbounds/remote_config.py

@@ -86,14 +86,14 @@ def init_config() -> Dict[str, str]:
                                  command with the string provided in the Outerbounds dashboard"
         )
 
-    # users still have a legacy format and that's ok.
-    if OBP_REMOTE_CONFIG_KEY not in remote_config:
-        return remote_config
-
     perimeter_config_url = get_perimeter_config_url_if_set_in_ob_config()
     if perimeter_config_url:
         remote_config[OBP_REMOTE_CONFIG_KEY] = perimeter_config_url
 
+    # users still have a legacy format and that's ok.
+    if OBP_REMOTE_CONFIG_KEY not in remote_config:
+        return remote_config
+
     metaflow_config = resolve_config_from_remote(
         remote_url=remote_config[OBP_REMOTE_CONFIG_KEY],
         auth_token=remote_config[AUTH_KEY],

{ob_metaflow_extensions-1.1.70.dist-info → ob_metaflow_extensions-1.1.71.dist-info}/METADATA

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

{ob_metaflow_extensions-1.1.70.dist-info → ob_metaflow_extensions-1.1.71.dist-info}/RECORD

@@ -1,9 +1,9 @@
 metaflow_extensions/outerbounds/__init__.py,sha256=TRGvIUMjkfneWtYUFSWoubu_Kf2ekAL4WLbV3IxOj9k,499
-metaflow_extensions/outerbounds/remote_config.py,sha256=HPFH4e3ZK3p-wS5HlS75fhR8_2avdD1AHQIZl2KnjeQ,4059
+metaflow_extensions/outerbounds/remote_config.py,sha256=Zpfpjgz68_ZgxlXezjzlsDLo4840rkWuZgwDB_5H57U,4059
 metaflow_extensions/outerbounds/config/__init__.py,sha256=mYo95obHU1IE1wbPkeVz_pfTzNqlNabp1QBEMTGllbE,112
 metaflow_extensions/outerbounds/plugins/__init__.py,sha256=46NgbJBhVowDR6FyQrZPF2jHHqRTSyCBCYIQAyQ4Ryo,9516
 metaflow_extensions/outerbounds/plugins/auth_server.py,sha256=JhlMFcR7SPSfR1C9w6GlqJq-NYNhOfISmHl2PdkYUok,2212
-metaflow_extensions/outerbounds/plugins/perimeters.py,sha256=z8tSAkWtiITB-JtSQS7fkhlBwvxSxeTgEwFjahAzv-U,2238
+metaflow_extensions/outerbounds/plugins/perimeters.py,sha256=QXh3SFP7GQbS-RAIxUOPbhPzQ7KDFVxZkTdKqFKgXjI,2697
 metaflow_extensions/outerbounds/plugins/kubernetes/__init__.py,sha256=5zG8gShSj8m7rgF4xgWBZFuY3GDP5n1T0ktjRpGJLHA,69
 metaflow_extensions/outerbounds/plugins/kubernetes/kubernetes_client.py,sha256=gj6Iaz26bGbZm3aQuNS18Mqh_80iJp5PgFwFSlJRcn8,1968
 metaflow_extensions/outerbounds/plugins/nim/__init__.py,sha256=GVnvSTjqYVj5oG2yh8KJFt7iZ33cEadDD5HbdmC9hJ0,1457
@@ -19,7 +19,7 @@ metaflow_extensions/outerbounds/toplevel/global_aliases_for_metaflow_package.py,
 metaflow_extensions/outerbounds/toplevel/plugins/azure/__init__.py,sha256=WUuhz2YQfI4fz7nIcipwwWq781eaoHEk7n4GAn1npDg,63
 metaflow_extensions/outerbounds/toplevel/plugins/gcp/__init__.py,sha256=BbZiaH3uILlEZ6ntBLKeNyqn3If8nIXZFq_Apd7Dhco,70
 metaflow_extensions/outerbounds/toplevel/plugins/kubernetes/__init__.py,sha256=5zG8gShSj8m7rgF4xgWBZFuY3GDP5n1T0ktjRpGJLHA,69
-ob_metaflow_extensions-1.1.70.dist-info/METADATA,sha256=oh54d2W1t23zvb-nWBXXp5CiCTn2wRpUKVrtgWERn5c,519
-ob_metaflow_extensions-1.1.70.dist-info/WHEEL,sha256=bb2Ot9scclHKMOLDEHY6B2sicWOgugjFKaJsT7vwMQo,110
-ob_metaflow_extensions-1.1.70.dist-info/top_level.txt,sha256=NwG0ukwjygtanDETyp_BUdtYtqIA_lOjzFFh1TsnxvI,20
-ob_metaflow_extensions-1.1.70.dist-info/RECORD,,
+ob_metaflow_extensions-1.1.71.dist-info/METADATA,sha256=GvyeQsYpp05xKshOovytjvOPBo1JbfmfyzhcXjBIAu0,519
+ob_metaflow_extensions-1.1.71.dist-info/WHEEL,sha256=bb2Ot9scclHKMOLDEHY6B2sicWOgugjFKaJsT7vwMQo,110
+ob_metaflow_extensions-1.1.71.dist-info/top_level.txt,sha256=NwG0ukwjygtanDETyp_BUdtYtqIA_lOjzFFh1TsnxvI,20
+ob_metaflow_extensions-1.1.71.dist-info/RECORD,,