ob-metaflow-extensions 1.4.0__py2.py3-none-any.whl → 1.4.1__py2.py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.

This version of ob-metaflow-extensions might be problematic. Click here for more details.

Files changed (5) hide show
  1. metaflow_extensions/outerbounds/remote_config.py +8 -3
  2. {ob_metaflow_extensions-1.4.0.dist-info → ob_metaflow_extensions-1.4.1.dist-info}/METADATA +1 -1
  3. {ob_metaflow_extensions-1.4.0.dist-info → ob_metaflow_extensions-1.4.1.dist-info}/RECORD +5 -5
  4. {ob_metaflow_extensions-1.4.0.dist-info → ob_metaflow_extensions-1.4.1.dist-info}/WHEEL +0 -0
  5. {ob_metaflow_extensions-1.4.0.dist-info → ob_metaflow_extensions-1.4.1.dist-info}/top_level.txt +0 -0
metaflow_extensions/outerbounds/remote_config.py CHANGED
@@ -11,6 +11,11 @@ from metaflow_extensions.outerbounds.plugins.perimeters import (
11
11
  get_perimeter_config_url_if_set_in_ob_config,
12
12
  )
13
13
 
14
+
15
+ class OuterboundsConfigException(MetaflowException):
16
+ _OB_CONFIG_EXCEPTION = True
17
+
18
+
14
19
  OBP_REMOTE_CONFIG_KEY = "OBP_METAFLOW_CONFIG_URL"
15
20
  HOSTNAME_KEY = "OBP_API_SERVER"
16
21
  AUTH_KEY = "METAFLOW_SERVICE_AUTH_KEY"
@@ -31,7 +36,7 @@ def read_config_from_local() -> Optional[Path]:
31
36
 
32
37
  # we should error because the user wants a specific config
33
38
  if profile:
34
- raise MetaflowException(
39
+ raise OuterboundsConfigException(
35
40
  f"Unable to locate METAFLOW_PROFILE {profile} in {config_path}"
36
41
  )
37
42
 
@@ -55,7 +60,7 @@ def resolve_config_from_remote(remote_url: str, auth_token: str) -> Dict[str, st
55
60
  data = response.json()
56
61
  return data["config"]
57
62
  except HTTPError:
58
- raise MetaflowException(
63
+ raise OuterboundsConfigException(
59
64
  "Error fetching resolving configuration. Make sure you have run \
60
65
  `outerbounds configure` with the correct value"
61
66
  )
@@ -81,7 +86,7 @@ def init_config() -> Dict[str, str]:
81
86
  try:
82
87
  remote_config = json.loads(config_path.read_text())
83
88
  except ValueError:
84
- raise MetaflowException(
89
+ raise OuterboundsConfigException(
85
90
  "Error decoding your metaflow config. Please run the `outerbounds configure` \
86
91
  command with the string provided in the Outerbounds dashboard"
87
92
  )
{ob_metaflow_extensions-1.4.0.dist-info → ob_metaflow_extensions-1.4.1.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: ob-metaflow-extensions
3
- Version: 1.4.0
3
+ Version: 1.4.1
4
4
  Summary: Outerbounds Platform Extensions for Metaflow
5
5
  Author: Outerbounds, Inc.
6
6
  License: Commercial
{ob_metaflow_extensions-1.4.0.dist-info → ob_metaflow_extensions-1.4.1.dist-info}/RECORD RENAMED
@@ -1,5 +1,5 @@
1
1
  metaflow_extensions/outerbounds/__init__.py,sha256=Gb8u06s9ClQsA_vzxmkCzuMnigPy7kKcDnLfb7eB-64,514
2
- metaflow_extensions/outerbounds/remote_config.py,sha256=pEFJuKDYs98eoB_-ryPjVi9b_c4gpHMdBHE14ltoxIU,4672
2
+ metaflow_extensions/outerbounds/remote_config.py,sha256=NIKiq-hrThOll7RLw2JfSN7K5Wz0QtjtRb5AVlHZ1vA,4787
3
3
  metaflow_extensions/outerbounds/config/__init__.py,sha256=JsQGRuGFz28fQWjUvxUgR8EKBLGRdLUIk_buPLJplJY,1225
4
4
  metaflow_extensions/outerbounds/plugins/__init__.py,sha256=ISFPm4cn8IvfZnM6Q4LqDalycr02v6gMsCPo9mRNSwk,14028
5
5
  metaflow_extensions/outerbounds/plugins/auth_server.py,sha256=_Q9_2EL0Xy77bCRphkwT1aSu8gQXRDOH-Z-RxTUO8N4,2202
@@ -124,7 +124,7 @@ metaflow_extensions/outerbounds/toplevel/plugins/ollama/__init__.py,sha256=GRSz2
124
124
  metaflow_extensions/outerbounds/toplevel/plugins/snowflake/__init__.py,sha256=LptpH-ziXHrednMYUjIaosS1SXD3sOtF_9_eRqd8SJw,50
125
125
  metaflow_extensions/outerbounds/toplevel/plugins/torchtune/__init__.py,sha256=uTVkdSk3xZ7hEKYfdlyVteWj5KeDwaM1hU9WT-_YKfI,50
126
126
  metaflow_extensions/outerbounds/toplevel/plugins/vllm/__init__.py,sha256=ekcgD3KVydf-a0xMI60P4uy6ePkSEoFHiGnDq1JM940,45
127
- ob_metaflow_extensions-1.4.0.dist-info/METADATA,sha256=2BzNOav9XSB5ff7cKCT-z-H_E72u4f3TTFRDws9h8C0,518
128
- ob_metaflow_extensions-1.4.0.dist-info/WHEEL,sha256=bb2Ot9scclHKMOLDEHY6B2sicWOgugjFKaJsT7vwMQo,110
129
- ob_metaflow_extensions-1.4.0.dist-info/top_level.txt,sha256=NwG0ukwjygtanDETyp_BUdtYtqIA_lOjzFFh1TsnxvI,20
130
- ob_metaflow_extensions-1.4.0.dist-info/RECORD,,
127
+ ob_metaflow_extensions-1.4.1.dist-info/METADATA,sha256=Sooyiz18IJSOTg_KU9kR7YGgj3rgQPs8Hzu0aP2fwXQ,518
128
+ ob_metaflow_extensions-1.4.1.dist-info/WHEEL,sha256=bb2Ot9scclHKMOLDEHY6B2sicWOgugjFKaJsT7vwMQo,110
129
+ ob_metaflow_extensions-1.4.1.dist-info/top_level.txt,sha256=NwG0ukwjygtanDETyp_BUdtYtqIA_lOjzFFh1TsnxvI,20
130
+ ob_metaflow_extensions-1.4.1.dist-info/RECORD,,