ob-metaflow-extensions 1.1.172__py2.py3-none-any.whl → 1.1.173__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/plugins/torchtune/__init__.py +4 -0
  2. {ob_metaflow_extensions-1.1.172.dist-info → ob_metaflow_extensions-1.1.173.dist-info}/METADATA +1 -1
  3. {ob_metaflow_extensions-1.1.172.dist-info → ob_metaflow_extensions-1.1.173.dist-info}/RECORD +5 -5
  4. {ob_metaflow_extensions-1.1.172.dist-info → ob_metaflow_extensions-1.1.173.dist-info}/WHEEL +0 -0
  5. {ob_metaflow_extensions-1.1.172.dist-info → ob_metaflow_extensions-1.1.173.dist-info}/top_level.txt +0 -0
metaflow_extensions/outerbounds/plugins/torchtune/__init__.py CHANGED
@@ -13,12 +13,14 @@ class TorchTune:
13
13
  def __init__(
14
14
  self,
15
15
  use_multi_node_config: bool = False,
16
+ config_overrides: Optional[Dict] = None,
16
17
  ):
17
18
  """
18
19
  Initialize the Tune launcher.
19
20
 
20
21
  :param use_multi_node_config: If True, attempt to build a distributed configuration
21
22
  from current.torch.torchrun_args.
23
+ :param config_overrides: Optional dictionary of config overrides for tune run.
22
24
  """
23
25
  self.multi_node_config = {}
24
26
  if use_multi_node_config:
@@ -37,6 +39,8 @@ class TorchTune:
37
39
  "num_processes": current.torch.torchrun_args["nproc_per_node"]
38
40
  * current.torch.torchrun_args["nnodes"],
39
41
  }
42
+ if config_overrides:
43
+ self.multi_node_config.update(config_overrides)
40
44
  print(
41
45
  f"[Metaflow Tune] Discovered multi-node config for torchrun: {self.multi_node_config}"
42
46
  )
{ob_metaflow_extensions-1.1.172.dist-info → ob_metaflow_extensions-1.1.173.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: ob-metaflow-extensions
3
- Version: 1.1.172
3
+ Version: 1.1.173
4
4
  Summary: Outerbounds Platform Extensions for Metaflow
5
5
  Author: Outerbounds, Inc.
6
6
  License: Commercial
{ob_metaflow_extensions-1.1.172.dist-info → ob_metaflow_extensions-1.1.173.dist-info}/RECORD RENAMED
@@ -66,7 +66,7 @@ metaflow_extensions/outerbounds/plugins/snowpark/snowpark_exceptions.py,sha256=F
66
66
  metaflow_extensions/outerbounds/plugins/snowpark/snowpark_job.py,sha256=aQphxX6jqYgfa83w387pEWl0keuLm38V53I8P8UL2ck,6887
67
67
  metaflow_extensions/outerbounds/plugins/snowpark/snowpark_service_spec.py,sha256=AI_kcm1hZV3JRxJkookcH6twiGnAYjk9Dx-MeoYz60Y,8511
68
68
  metaflow_extensions/outerbounds/plugins/tensorboard/__init__.py,sha256=9lUM4Cqi5RjrHBRfG6AQMRz8-R96eZC8Ih0KD2lv22Y,1858
69
- metaflow_extensions/outerbounds/plugins/torchtune/__init__.py,sha256=TOXNeyhcgd8VxplXO_oEuryFEsbk0tikn5GL0-44SU8,5853
69
+ metaflow_extensions/outerbounds/plugins/torchtune/__init__.py,sha256=Psj2ybj_E1qp5KK2inon9e4ZecaRxnPtW3ngcirbO2g,6094
70
70
  metaflow_extensions/outerbounds/plugins/vllm/__init__.py,sha256=O04DPVoEdCZhPbvdldaE4ztoAxJNXU-ExosBCqe43v8,6463
71
71
  metaflow_extensions/outerbounds/plugins/vllm/constants.py,sha256=ODX_uM5iYrzpVltsAdSf9Jo0DAOMiZ3647DcKdCnlS0,24
72
72
  metaflow_extensions/outerbounds/plugins/vllm/exceptions.py,sha256=8m65k2L17zXgSkgU299DWqxr1wGUMsZgSJw0hBRizJ0,49
@@ -84,7 +84,7 @@ metaflow_extensions/outerbounds/toplevel/plugins/ollama/__init__.py,sha256=GRSz2
84
84
  metaflow_extensions/outerbounds/toplevel/plugins/snowflake/__init__.py,sha256=LptpH-ziXHrednMYUjIaosS1SXD3sOtF_9_eRqd8SJw,50
85
85
  metaflow_extensions/outerbounds/toplevel/plugins/torchtune/__init__.py,sha256=uTVkdSk3xZ7hEKYfdlyVteWj5KeDwaM1hU9WT-_YKfI,50
86
86
  metaflow_extensions/outerbounds/toplevel/plugins/vllm/__init__.py,sha256=ekcgD3KVydf-a0xMI60P4uy6ePkSEoFHiGnDq1JM940,45
87
- ob_metaflow_extensions-1.1.172.dist-info/METADATA,sha256=AM5dt47pVeIjEA9pwZzrq_o1FF5STn8KrRWecu1J_AI,521
88
- ob_metaflow_extensions-1.1.172.dist-info/WHEEL,sha256=bb2Ot9scclHKMOLDEHY6B2sicWOgugjFKaJsT7vwMQo,110
89
- ob_metaflow_extensions-1.1.172.dist-info/top_level.txt,sha256=NwG0ukwjygtanDETyp_BUdtYtqIA_lOjzFFh1TsnxvI,20
90
- ob_metaflow_extensions-1.1.172.dist-info/RECORD,,
87
+ ob_metaflow_extensions-1.1.173.dist-info/METADATA,sha256=um9vZD7UiAvC1J4Jysgp84JvWxtEj8Z7KNyqVGhVdDQ,521
88
+ ob_metaflow_extensions-1.1.173.dist-info/WHEEL,sha256=bb2Ot9scclHKMOLDEHY6B2sicWOgugjFKaJsT7vwMQo,110
89
+ ob_metaflow_extensions-1.1.173.dist-info/top_level.txt,sha256=NwG0ukwjygtanDETyp_BUdtYtqIA_lOjzFFh1TsnxvI,20
90
+ ob_metaflow_extensions-1.1.173.dist-info/RECORD,,