ob-metaflow-extensions 1.1.152__py2.py3-none-any.whl → 1.1.154__py2.py3-none-any.whl

metaflow_extensions/outerbounds/__init__.py

@@ -1,6 +1,6 @@
 import metaflow.metaflow_config_funcs
 
-from metaflow_extensions.outerbounds.remote_config import init_config
+from metaflow_extensions.outerbounds.remote_config import init_config, reload_config
 
 # we want to overide OSS Metaflow's initialization behavior with our own to support remote configs
 # we're reassigning the METAFLOW_CONFIG variable because all downstream settings rely on it and

metaflow_extensions/outerbounds/remote_config.py

@@ -112,3 +112,22 @@ def _init_debug(*args, **kwargs):
         init_str = "ob_extension_init:"
         kwargs["file"] = sys.stderr
         print(init_str, *args, **kwargs)
+
+
+def reload_config():
+    """
+    This function is used to reload the config. Currently its a best effort implementation
+    that will only reload auth token.
+    """
+    _init_debug("reloading config")
+    if CONFIG_READ_ONCE_KEY in os.environ:
+        del os.environ[CONFIG_READ_ONCE_KEY]
+    config = init_config()
+    import metaflow.metaflow_config
+
+    metaflow.metaflow_config.SERVICE_AUTH_KEY = config.get(AUTH_KEY)
+    metaflow.metaflow_config.SERVICE_HEADERS["x-api-key"] = config.get(AUTH_KEY)
+    if config:
+        _init_debug("reloaded config")
+    else:
+        _init_debug("no config to reload")

{ob_metaflow_extensions-1.1.152.dist-info → ob_metaflow_extensions-1.1.154.dist-info}/METADATA

@@ -1,13 +1,13 @@
 Metadata-Version: 2.1
 Name: ob-metaflow-extensions
-Version: 1.1.152
+Version: 1.1.154
 Summary: Outerbounds Platform Extensions for Metaflow
 Author: Outerbounds, Inc.
 License: Commercial
 Description-Content-Type: text/markdown
 Requires-Dist: boto3
 Requires-Dist: kubernetes
-Requires-Dist: ob-metaflow (==2.15.11.1)
+Requires-Dist: ob-metaflow (==2.15.11.2)
 
 # Outerbounds platform package
 

{ob_metaflow_extensions-1.1.152.dist-info → ob_metaflow_extensions-1.1.154.dist-info}/RECORD

@@ -1,5 +1,5 @@
-metaflow_extensions/outerbounds/__init__.py,sha256=TRGvIUMjkfneWtYUFSWoubu_Kf2ekAL4WLbV3IxOj9k,499
-metaflow_extensions/outerbounds/remote_config.py,sha256=Zpfpjgz68_ZgxlXezjzlsDLo4840rkWuZgwDB_5H57U,4059
+metaflow_extensions/outerbounds/__init__.py,sha256=Gb8u06s9ClQsA_vzxmkCzuMnigPy7kKcDnLfb7eB-64,514
+metaflow_extensions/outerbounds/remote_config.py,sha256=pEFJuKDYs98eoB_-ryPjVi9b_c4gpHMdBHE14ltoxIU,4672
 metaflow_extensions/outerbounds/config/__init__.py,sha256=JsQGRuGFz28fQWjUvxUgR8EKBLGRdLUIk_buPLJplJY,1225
 metaflow_extensions/outerbounds/plugins/__init__.py,sha256=eHcM_t2Mzlge7B9Dv3VGVM5x8qNZYdLyqBOAC6uRxec,13228
 metaflow_extensions/outerbounds/plugins/auth_server.py,sha256=_Q9_2EL0Xy77bCRphkwT1aSu8gQXRDOH-Z-RxTUO8N4,2202
@@ -68,7 +68,7 @@ metaflow_extensions/outerbounds/toplevel/plugins/gcp/__init__.py,sha256=BbZiaH3u
 metaflow_extensions/outerbounds/toplevel/plugins/kubernetes/__init__.py,sha256=5zG8gShSj8m7rgF4xgWBZFuY3GDP5n1T0ktjRpGJLHA,69
 metaflow_extensions/outerbounds/toplevel/plugins/ollama/__init__.py,sha256=GRSz2zwqkvlmFS6bcfYD_CX6CMko9DHQokMaH1iBshA,47
 metaflow_extensions/outerbounds/toplevel/plugins/snowflake/__init__.py,sha256=LptpH-ziXHrednMYUjIaosS1SXD3sOtF_9_eRqd8SJw,50
-ob_metaflow_extensions-1.1.152.dist-info/METADATA,sha256=gOiQAMbmuGwKu2qJJoWmaiorTfxt_9qxu0eVeiWJTRE,521
-ob_metaflow_extensions-1.1.152.dist-info/WHEEL,sha256=bb2Ot9scclHKMOLDEHY6B2sicWOgugjFKaJsT7vwMQo,110
-ob_metaflow_extensions-1.1.152.dist-info/top_level.txt,sha256=NwG0ukwjygtanDETyp_BUdtYtqIA_lOjzFFh1TsnxvI,20
-ob_metaflow_extensions-1.1.152.dist-info/RECORD,,
+ob_metaflow_extensions-1.1.154.dist-info/METADATA,sha256=EEzzuRx6dYL7E74RUnGyd96Zor8SdgeYjjhoAALDBYY,521
+ob_metaflow_extensions-1.1.154.dist-info/WHEEL,sha256=bb2Ot9scclHKMOLDEHY6B2sicWOgugjFKaJsT7vwMQo,110
+ob_metaflow_extensions-1.1.154.dist-info/top_level.txt,sha256=NwG0ukwjygtanDETyp_BUdtYtqIA_lOjzFFh1TsnxvI,20
+ob_metaflow_extensions-1.1.154.dist-info/RECORD,,