ob-metaflow-extensions 1.1.172__tar.gz → 1.1.173__tar.gz

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 (95) hide show
  1. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/PKG-INFO +1 -1
  2. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/torchtune/__init__.py +4 -0
  3. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/ob_metaflow_extensions.egg-info/PKG-INFO +1 -1
  4. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/setup.py +1 -1
  5. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/README.md +0 -0
  6. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/__init__.py +0 -0
  7. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/config/__init__.py +0 -0
  8. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/__init__.py +0 -0
  9. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/apps/__init__.py +0 -0
  10. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/apps/app_utils.py +0 -0
  11. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/apps/consts.py +0 -0
  12. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/apps/deploy_decorator.py +0 -0
  13. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/apps/supervisord_utils.py +0 -0
  14. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/auth_server.py +0 -0
  15. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/aws/__init__.py +0 -0
  16. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/aws/assume_role.py +0 -0
  17. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/aws/assume_role_decorator.py +0 -0
  18. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/card_utilities/__init__.py +0 -0
  19. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/card_utilities/async_cards.py +0 -0
  20. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/card_utilities/extra_components.py +0 -0
  21. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/card_utilities/injector.py +0 -0
  22. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/checkpoint_datastores/__init__.py +0 -0
  23. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/checkpoint_datastores/coreweave.py +0 -0
  24. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/checkpoint_datastores/nebius.py +0 -0
  25. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/fast_bakery/__init__.py +0 -0
  26. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/fast_bakery/docker_environment.py +0 -0
  27. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/fast_bakery/fast_bakery.py +0 -0
  28. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/fast_bakery/fast_bakery_cli.py +0 -0
  29. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/fast_bakery/fast_bakery_decorator.py +0 -0
  30. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/kubernetes/__init__.py +0 -0
  31. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/kubernetes/kubernetes_client.py +0 -0
  32. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/kubernetes/pod_killer.py +0 -0
  33. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/nim/card.py +0 -0
  34. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/nim/nim_decorator.py +0 -0
  35. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/nim/nim_manager.py +0 -0
  36. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/nim/utils.py +0 -0
  37. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/nvcf/__init__.py +0 -0
  38. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/nvcf/constants.py +0 -0
  39. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/nvcf/exceptions.py +0 -0
  40. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/nvcf/heartbeat_store.py +0 -0
  41. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/nvcf/nvcf.py +0 -0
  42. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/nvcf/nvcf_cli.py +0 -0
  43. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/nvcf/nvcf_decorator.py +0 -0
  44. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/nvcf/utils.py +0 -0
  45. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/nvct/__init__.py +0 -0
  46. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/nvct/exceptions.py +0 -0
  47. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/nvct/nvct.py +0 -0
  48. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/nvct/nvct_cli.py +0 -0
  49. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/nvct/nvct_decorator.py +0 -0
  50. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/nvct/nvct_runner.py +0 -0
  51. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/nvct/utils.py +0 -0
  52. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/ollama/__init__.py +0 -0
  53. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/ollama/constants.py +0 -0
  54. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/ollama/exceptions.py +0 -0
  55. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/ollama/ollama.py +0 -0
  56. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/ollama/status_card.py +0 -0
  57. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/perimeters.py +0 -0
  58. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/profilers/deco_injector.py +0 -0
  59. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/profilers/gpu_profile_decorator.py +0 -0
  60. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/secrets/__init__.py +0 -0
  61. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/secrets/secrets.py +0 -0
  62. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/snowflake/__init__.py +0 -0
  63. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/snowflake/snowflake.py +0 -0
  64. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/snowpark/__init__.py +0 -0
  65. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/snowpark/snowpark.py +0 -0
  66. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/snowpark/snowpark_cli.py +0 -0
  67. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/snowpark/snowpark_client.py +0 -0
  68. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/snowpark/snowpark_decorator.py +0 -0
  69. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/snowpark/snowpark_exceptions.py +0 -0
  70. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/snowpark/snowpark_job.py +0 -0
  71. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/snowpark/snowpark_service_spec.py +0 -0
  72. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/tensorboard/__init__.py +0 -0
  73. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/vllm/__init__.py +0 -0
  74. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/vllm/constants.py +0 -0
  75. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/vllm/exceptions.py +0 -0
  76. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/vllm/status_card.py +0 -0
  77. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/vllm/vllm_manager.py +0 -0
  78. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/profilers/__init__.py +0 -0
  79. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/profilers/gpu.py +0 -0
  80. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/remote_config.py +0 -0
  81. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/toplevel/__init__.py +0 -0
  82. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/toplevel/global_aliases_for_metaflow_package.py +0 -0
  83. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/toplevel/ob_internal.py +0 -0
  84. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/toplevel/plugins/azure/__init__.py +0 -0
  85. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/toplevel/plugins/gcp/__init__.py +0 -0
  86. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/toplevel/plugins/kubernetes/__init__.py +0 -0
  87. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/toplevel/plugins/ollama/__init__.py +0 -0
  88. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/toplevel/plugins/snowflake/__init__.py +0 -0
  89. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/toplevel/plugins/torchtune/__init__.py +0 -0
  90. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/toplevel/plugins/vllm/__init__.py +0 -0
  91. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/ob_metaflow_extensions.egg-info/SOURCES.txt +0 -0
  92. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/ob_metaflow_extensions.egg-info/dependency_links.txt +0 -0
  93. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/ob_metaflow_extensions.egg-info/requires.txt +0 -0
  94. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/ob_metaflow_extensions.egg-info/top_level.txt +0 -0
  95. {ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/setup.cfg +0 -0
{ob_metaflow_extensions-1.1.172 → ob_metaflow_extensions-1.1.173}/PKG-INFO 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 → ob_metaflow_extensions-1.1.173}/metaflow_extensions/outerbounds/plugins/torchtune/__init__.py RENAMED
@@ -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 → ob_metaflow_extensions-1.1.173}/ob_metaflow_extensions.egg-info/PKG-INFO 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 → ob_metaflow_extensions-1.1.173}/setup.py RENAMED
@@ -2,7 +2,7 @@ from setuptools import setup, find_namespace_packages
2
2
  from pathlib import Path
3
3
 
4
4
 
5
- version = "1.1.172"
5
+ version = "1.1.173"
6
6
  this_directory = Path(__file__).parent
7
7
  long_description = (this_directory / "README.md").read_text()
8
8