torchx-nightly 2026.1.11__py3-none-any.whl → 2026.1.22__py3-none-any.whl

torchx/cli/cmd_list.py

@@ -11,7 +11,6 @@ import argparse
 import logging
 
 from tabulate import tabulate
-
 from torchx.cli.cmd_base import SubCommand
 from torchx.runner import get_runner
 from torchx.schedulers import get_default_scheduler_name, get_scheduler_factories

torchx/cli/cmd_log.py

@@ -27,7 +27,6 @@ from torchx.util.log_tee_helpers import (
     _find_role_replicas as find_role_replicas,
     _prefix_line,
 )
-
 from torchx.util.types import none_throws
 
 logger: logging.Logger = logging.getLogger(__name__)

torchx/cli/cmd_tracker.py

@@ -10,7 +10,6 @@ import argparse
 import logging
 
 from tabulate import tabulate
-
 from torchx.cli.cmd_base import SubCommand
 from torchx.runner.api import get_configured_trackers
 from torchx.tracker.api import build_trackers, TrackerBase

torchx/components/component_test_base.py

@@ -25,9 +25,7 @@ from types import ModuleType
 from typing import Any, Callable, Dict, Optional
 
 from torchx.runner import get_runner
-
 from torchx.specs import AppDef, AppStatus
-
 from torchx.specs.builders import _create_args_parser
 from torchx.specs.finder import get_component
 

torchx/components/integration_tests/integ_tests.py

@@ -18,7 +18,6 @@ from torchx.cli.cmd_log import get_logs
 from torchx.components.integration_tests.component_provider import ComponentProvider
 from torchx.runner import get_runner
 from torchx.specs import AppHandle, AppState, AppStatus, CfgVal
-
 from torchx.util.types import none_throws
 
 

torchx/distributed/__init__.py

@@ -17,7 +17,6 @@ from typing import Any, Iterator
 import torch
 import torch.distributed as dist
 from torch.distributed.distributed_c10d import _get_default_group
-
 from torchx.util.cuda import has_cuda_devices
 from typing_extensions import Literal
 

torchx/examples/apps/lightning/profiler.py

@@ -20,7 +20,6 @@ import time
 from typing import Dict
 
 from pytorch_lightning.loggers.logger import Logger
-
 from pytorch_lightning.profilers.profiler import Profiler
 
 

torchx/runner/api.py

@@ -52,7 +52,6 @@ from torchx.tracker.api import (
     tracker_config_env_var_name,
 )
 from torchx.util.session import get_session_id_or_create_new, TORCHX_INTERNAL_SESSION_ID
-
 from torchx.util.types import none_throws
 from torchx.workspace import WorkspaceMixin
 

torchx/schedulers/aws_batch_scheduler.py

@@ -66,7 +66,6 @@ from torchx.schedulers.api import (
     Scheduler,
     Stream,
 )
-
 from torchx.schedulers.devices import get_device_mounts
 from torchx.schedulers.ids import make_unique
 from torchx.specs.api import (
@@ -188,7 +187,7 @@ def resource_requirements_from_resource(resource: Resource) -> List[Dict[str, st
 
 
 def resource_from_resource_requirements(
-    resource_requirements: List[Dict[str, str]]
+    resource_requirements: List[Dict[str, str]],
 ) -> Resource:
     resrc_req = {
         ResourceType.from_str(r["type"]): int(r["value"]) for r in resource_requirements

torchx/schedulers/aws_sagemaker_scheduler.py

@@ -31,7 +31,6 @@ from typing import (
 
 import boto3
 import yaml
-
 from sagemaker.pytorch import PyTorch
 from torchx.components.structured_arg import StructuredNameArgument
 from torchx.schedulers.api import (

torchx/schedulers/kubernetes_mcad_scheduler.py

@@ -32,7 +32,6 @@ Learn more about running distributed trainers :py:mod:`torchx.components.dist`
 import json
 import logging
 import re
-
 import warnings
 from dataclasses import dataclass
 from datetime import datetime
@@ -77,7 +76,6 @@ from torchx.specs.api import (
     runopts,
     VolumeMount,
 )
-
 from torchx.workspace.docker_workspace import DockerWorkspaceMixin
 
 if TYPE_CHECKING:

torchx/schedulers/local_scheduler.py

@@ -55,7 +55,6 @@ from torchx.schedulers.ids import make_unique
 from torchx.schedulers.streams import Tee
 from torchx.specs import AppDryRunInfo
 from torchx.specs.api import AppDef, AppState, is_terminal, macros, NONE, Role, runopts
-
 from torchx.util.types import none_throws
 
 log: logging.Logger = logging.getLogger(__name__)

torchx/specs/__init__.py

@@ -12,7 +12,6 @@ used by components to define the apps which can then be launched via a TorchX
 scheduler or pipeline adapter.
 """
 import difflib
-
 import os
 from typing import Callable, Dict, Iterator, Mapping, Optional
 
@@ -50,9 +49,7 @@ from torchx.specs.api import (
     Workspace,
 )
 from torchx.specs.builders import make_app_handle, materialize_appdef, parse_mounts
-
 from torchx.util.entrypoints import load_group
-
 from torchx.util.modules import import_attr
 
 GiB: int = 1024

torchx/specs/finder.py

@@ -20,7 +20,6 @@ from types import ModuleType
 from typing import Callable, Dict, Generator, List, Optional, Union
 
 from torchx.specs import AppDef
-
 from torchx.specs.file_linter import (
     ComponentFunctionValidator,
     get_fn_docstring,

torchx/tracker/api.py

@@ -179,7 +179,7 @@ def _extract_tracker_name_and_config_from_environ() -> Mapping[str, Optional[str
 
 
 def build_trackers(
-    factory_and_config: Mapping[str, Optional[str]]
+    factory_and_config: Mapping[str, Optional[str]],
 ) -> Iterable[TrackerBase]:
     trackers = []
 

torchx/tracker/backend/fsspec.py

@@ -16,7 +16,6 @@ from dataclasses import dataclass
 from typing import Any, Dict, Iterable, Mapping, Optional
 
 import fsspec
-
 from torchx.tracker.api import Lineage, TrackerArtifact, TrackerBase, TrackerSource
 
 

torchx/tracker/mlflow.py

@@ -16,7 +16,6 @@ from typing import Any, Dict, Iterable, Mapping, Optional, Sequence
 import mlflow
 from mlflow import MlflowClient
 from mlflow.entities import Experiment, Run
-
 from torchx.distributed import on_rank0_first
 from torchx.runner.config import get_configs
 from torchx.tracker.api import (

torchx/workspace/docker_workspace.py

@@ -16,7 +16,6 @@ import tempfile
 from typing import Dict, IO, Iterable, Mapping, Optional, TextIO, Tuple, TYPE_CHECKING
 
 import fsspec
-
 import torchx
 from docker.errors import BuildError
 from torchx.specs import AppDef, CfgVal, Role, runopts

{torchx_nightly-2026.1.11.dist-info → torchx_nightly-2026.1.22.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: torchx-nightly
-Version: 2026.1.11
+Version: 2026.1.22
 Summary: TorchX SDK and Components
 Home-page: https://github.com/meta-pytorch/torchx
 Author: TorchX Devs

{torchx_nightly-2026.1.11.dist-info → torchx_nightly-2026.1.22.dist-info}/RECORD

@@ -16,16 +16,16 @@ torchx/cli/cmd_cancel.py,sha256=NKfOCu_44Lch9vliGSQ0Uv6BVqpUqj7Tob652TI-ua4,835
 torchx/cli/cmd_configure.py,sha256=1kTv0qbsbV44So74plAySwWu56pQrqjhfW_kbfdC3Rw,1722
 torchx/cli/cmd_delete.py,sha256=US1f6Jvyhz4R_0Q0a8GeNTDMrhzo8WE_ECcdOf0MjKE,835
 torchx/cli/cmd_describe.py,sha256=E5disbHoKTsqYKp2s3DaFW9GDLCCOgdOc3pQoHKoyCs,1283
-torchx/cli/cmd_list.py,sha256=alkS9aIaDI8lX3W8uj8Vtr3IU3G2VeCuokKSd3zOFug,1409
-torchx/cli/cmd_log.py,sha256=v-EZYUDOcG95rEgTnrsmPJMUyxM9Mk8YFAJtUxtgViE,5475
+torchx/cli/cmd_list.py,sha256=deu920UTFJFTNVBdgSXhgMUpbJF4G9-xNny6XIqU4KA,1408
+torchx/cli/cmd_log.py,sha256=nEzwVs1QwtrDFXtiAgPXnCu2YiBgVAtACIdpOSAYAU8,5474
 torchx/cli/cmd_run.py,sha256=z8wS-M2W9hHZfLkA6DFiV6Y0LFS9KfEBc_NTwAwdviQ,18780
 torchx/cli/cmd_runopts.py,sha256=NWZiP8XpQjfTDJgays2c6MgL_8wxFoeDge6NstaZdKk,1302
 torchx/cli/cmd_status.py,sha256=22IAEmKs0qkG6kJi83u9dRX2Q-ntT7yehVx7FxtY-vQ,2114
-torchx/cli/cmd_tracker.py,sha256=9gmOmYi-89qQRGQfSrXCTto7ve54_JKFqs_wa7oRUA8,5223
+torchx/cli/cmd_tracker.py,sha256=pWVqXGUiwPE5_aWPCn_j-ov2EQkH2f0Xdv5DZW5U3Tg,5222
 torchx/cli/colors.py,sha256=yLMes7e_UoLAfhxE0W6edhc58t83UHAlnCN2ANPeuXw,568
 torchx/cli/main.py,sha256=1DJTmKdvPW_7hod8OUVT3Br2uwsZVEDU-2bTE0NJ0zY,3559
 torchx/components/__init__.py,sha256=JaVte0j9Gqi6IrjZKudJ2Kr3gkdHsvlCdRTo-zYpSRo,11815
-torchx/components/component_test_base.py,sha256=22iNSdVa_qTW3SMM30Pw5UEWlK4DZVw0C03EqYiaLOI,4150
+torchx/components/component_test_base.py,sha256=2kIC7odZQwpsFRjdHW1m0_BY5Uh6IZlIOx0bWgLB_JI,4148
 torchx/components/dist.py,sha256=6DNPEvHVqEifmM8g1L7HVY169cQv_7tSfSlh3o6lTp4,14930
 torchx/components/interpret.py,sha256=g8gkKdDJvsBfX1ZrpVT7n2bMEtmwRV_1AqDyAnnQ_aA,697
 torchx/components/metrics.py,sha256=1gbp8BfzZWGa7PD1db5vRADlONzmae4qSBUUdCWayr0,2814
@@ -35,8 +35,8 @@ torchx/components/train.py,sha256=vtrQXRcD7bIcbb3lSeyD9BBlIe1mv1WNW6rnLK9R0Mw,12
 torchx/components/utils.py,sha256=IMjihhgs7nO67YtTetUBjN_CRpyIyyQsaJBkp7mpHfk,9368
 torchx/components/integration_tests/__init__.py,sha256=Md3cCHD7Ano9kV15PqGbicgUO-RMdh4aVy1yKiDt_xE,208
 torchx/components/integration_tests/component_provider.py,sha256=g-4ig1vtd5Vzgug0VAKRAFUt6KAV3TgQrBCrwRSJ7ZY,3981
-torchx/components/integration_tests/integ_tests.py,sha256=O8jd8Jq5O0mns7xzIFsHexBDHkIIAIfELQkWCzNPzRw,5165
-torchx/distributed/__init__.py,sha256=kh9YzDwWX7zFJJ8StR9qhMM2V3-66INs9i3ztDF-1ho,10252
+torchx/components/integration_tests/integ_tests.py,sha256=JrOAauk4xbB3bB_yf8yZl69ddTESdacEf9JrMHcoaJU,5164
+torchx/distributed/__init__.py,sha256=m0QXzwqpXyubk4g7JB79tHjT3Ab5JiVCQM7MRf5H9a0,10251
 torchx/examples/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 torchx/examples/torchx_out_of_sync_training.py,sha256=sXiI1G8aGsfuvxRdBszDgM8pSplqhgfXjRnAcgRwNGM,397
 torchx/examples/apps/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -46,11 +46,11 @@ torchx/examples/apps/lightning/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5
 torchx/examples/apps/lightning/data.py,sha256=kSv_DFqtFVkNjZ46HT7GApImc9lMD7liy929dUrFWwM,6610
 torchx/examples/apps/lightning/interpret.py,sha256=Hd3kE5a6FyhxCmJBfTzb4Tlj518zhX8V0XvZfzu4nqE,5256
 torchx/examples/apps/lightning/model.py,sha256=4CgObWfANqDN9emYSdmCpbRe_V_Lef_Hd3M-yayDbZE,4045
-torchx/examples/apps/lightning/profiler.py,sha256=SSSihnwjeUTkBoz0E3qn1b-wbkfUIowscx2ND_37zyw,1915
+torchx/examples/apps/lightning/profiler.py,sha256=ogL3mO4YGPebdCFckkeHX3BzJD3niU189DCnrBEoBGI,1914
 torchx/examples/apps/lightning/train.py,sha256=0wvvshGHvZowePB4LfclXwn40X7i9euM0ReETWBcPSo,6253
 torchx/pipelines/__init__.py,sha256=2MbRVk5xwRjg-d2qPemeXpEhDsocMQumPQ53lsesZAI,606
 torchx/runner/__init__.py,sha256=x8Sz7s_tLxPgJgvWIhK4ju9BNZU61uBFywGwDY6CqJs,315
-torchx/runner/api.py,sha256=Qi12Kjkr_zpQBesbLuCtgKET8JhHnQk22MV7Czi4l1A,30832
+torchx/runner/api.py,sha256=tN8087Hi7OHX1lVCmjccFgM1tcZwaxeJGMIvE4ZDrb4,30831
 torchx/runner/config.py,sha256=SaKOB50d79WaMFPWK8CC4as6UaNFaRGhrBkfajq3KC4,18311
 torchx/runner/events/__init__.py,sha256=cMiNjnr4eUNQ2Nxxtu4nsvN5lu56b-a6nJ-ct3i7DQk,5536
 torchx/runner/events/api.py,sha256=bvxKBAYK8LzbrBNaNLgL1x0aivtfANmWo1EMGOrSR8k,2668
@@ -60,22 +60,22 @@ torchx/runtime/tracking/__init__.py,sha256=dYnAPnrXYREfPXkpHhdOFkcYIODWEbA13PdD-
 torchx/runtime/tracking/api.py,sha256=SmUQyUKZqG3KlAhT7CJOGqRz1O274E4m63wQeOVq3CU,5472
 torchx/schedulers/__init__.py,sha256=FQN9boQM4mwOD3sK9LZ3GBgw-gJ7Vx4MFj6z6ATQIrc,2211
 torchx/schedulers/api.py,sha256=wT9H_ZTmpTHHweevDJbkV7NKXfwileHrt1bbhhCgj3c,16488
-torchx/schedulers/aws_batch_scheduler.py,sha256=b6xC4BQKb7zagOGS6_z3_6fmOLsSEOxSprkGUE-yfJE,29412
-torchx/schedulers/aws_sagemaker_scheduler.py,sha256=DnNF6huHGZLSUGWqKml4qGiWvmyDzX0i45tjsRfkedg,20881
+torchx/schedulers/aws_batch_scheduler.py,sha256=tsQmeqEBLR_Zcm7jWbbZnoZ5TFvo9FHhEt00LgZAnzM,29412
+torchx/schedulers/aws_sagemaker_scheduler.py,sha256=BRa85fqWcPK-B10cYMmm-CbJu0smxOsTXknAbOCfaYA,20880
 torchx/schedulers/devices.py,sha256=RjVcu22ZRl_9OKtOtmA1A3vNXgu2qD6A9ST0L0Hsg4I,1734
 torchx/schedulers/docker_scheduler.py,sha256=Kud3AIzQtMekgjlqcg1eNDb8kk29aPbGYOMAvPTZdhM,16840
 torchx/schedulers/ids.py,sha256=8Qhf1Xqh845mwL-RXnWZXqIILNvml3z8udEXPFpyO7U,2247
-torchx/schedulers/kubernetes_mcad_scheduler.py,sha256=FclJEdBdlgtBqKDbgd95oAk5Ya5XNTrwysfX7GS80GY,42896
+torchx/schedulers/kubernetes_mcad_scheduler.py,sha256=G2LZTNMEJRo34osBxMUScYXUG9fYi_Hak78-PH5cJUo,42894
 torchx/schedulers/kubernetes_scheduler.py,sha256=kYO08hqVlZtNe_FZQP_e8WQk1P8-8SVkXZuY3Zm_Znk,39640
-torchx/schedulers/local_scheduler.py,sha256=xGQbI02BNWGF91g00So6hCcYvR90bUAZ7fPzqnm3Ww8,41892
+torchx/schedulers/local_scheduler.py,sha256=Ga5nZ6mxqBa8KcD32UAgZiY7-uhHXnBAIhwNHilhEkw,41891
 torchx/schedulers/lsf_scheduler.py,sha256=vUvEJb02u7WI6y7DsWJxJFXNylRucU7FqkBX7xwLTak,17638
 torchx/schedulers/slurm_scheduler.py,sha256=ipDVDtgfqgL6c35NyoJgSPuQFt8-AeXVXAnXJVvmzrc,32032
 torchx/schedulers/streams.py,sha256=8_SLezgnWgfv_zXUsJCUM34-h2dtv25NmZuxEwkzmxw,2007
-torchx/specs/__init__.py,sha256=TaC0AveTebkCMo5hmdY1wGpo09vFDqzWnsT166ionTw,7108
+torchx/specs/__init__.py,sha256=tFvFg0uRwwZgZHiD3hfMDGlEpfu9SIZPWvCohEgqcvQ,7105
 torchx/specs/api.py,sha256=7FdLFfadNWqXTLJ_EtP5t1uVS2Vc_4Gj5GLFoI628oE,49338
 torchx/specs/builders.py,sha256=Ye3of4MupJ-da8vLaX6_-nzGo_FRw1BFpYsX6dAZCNk,13730
 torchx/specs/file_linter.py,sha256=z0c4mKJv47BWiPaWCdUM0A8kHwnj4b1s7oTmESuD9Tc,14407
-torchx/specs/finder.py,sha256=gWQNEFrLYqrZoI0gMMhQ70YAC4sxqS0ZFpoWAmcVi44,17438
+torchx/specs/finder.py,sha256=zBSjcywPO-BnYAUwG9EMi0_1UPBfEBNdA3C8WXz8KQU,17437
 torchx/specs/named_resources_aws.py,sha256=ZNAbw6lD8NUlMfcJ-LpX14dMSaHO7m4Yt9iHwAF44yg,11674
 torchx/specs/named_resources_generic.py,sha256=Sg4tAdqiiWDrDz2Lj_pnfsjzGIXKTou73wPseh6j55w,2646
 torchx/specs/overlays.py,sha256=HmY2yzC8ejgihviNWFT4rbYmP-gTcqpxVZTP6qBiIYM,3778
@@ -86,10 +86,10 @@ torchx/specs/test/components/a/b/c.py,sha256=FhixafzNqpS5zvggtWIWLxRd6HIxsOmct-d
 torchx/specs/test/components/c/__init__.py,sha256=5CBMckkpqJUdxBQBYHGSsItqq1gj2V0UiCw02Qfq6MM,246
 torchx/specs/test/components/c/d.py,sha256=2AjE-FmQXJTw3hws66O83ToQPmjOEZLDf-jDAKrrUkQ,546
 torchx/tracker/__init__.py,sha256=qo39aOa0Dz9zt4TtFkqPeIaH7MNqdAkFlGaOFiDLXTI,4375
-torchx/tracker/api.py,sha256=WZ7TYdbSVx_5h5MlX9EwQLRpxmIf0oKdiQwQ0zvkO3o,11262
-torchx/tracker/mlflow.py,sha256=poeoIXVPzr2sxgi515fMGRH83KAFNL6XFILMh0EQ2Dw,14487
+torchx/tracker/api.py,sha256=4rteINX8ZMv_03t75qOPU-rP3YeIPmm6N1HX9t8lVQg,11263
+torchx/tracker/mlflow.py,sha256=arl70oNw76VNIpf_gEP5p7A7OnDQVIfWZDEyImuM_Gc,14486
 torchx/tracker/backend/__init__.py,sha256=fE0IHi1JJpxsNVBNzWNee2thrNXFFRhY94c80RxNSIE,231
-torchx/tracker/backend/fsspec.py,sha256=528xKryBE27Rm_OHD7r2R6fmVAclknBtoy1s034Ny6c,10440
+torchx/tracker/backend/fsspec.py,sha256=1lJ1SoaTXl8ajvIJtp9pUmQgTRw7nF5D0Hv3susfYmE,10439
 torchx/util/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 torchx/util/cuda.py,sha256=-ZTa1WCLnY2WtSWAdWufLQqZSDCZfZsloBuiS84LIkU,1099
 torchx/util/datetime.py,sha256=hV6Sg0u5KTBe68yrmy_RGCC5su0i4Tb_mAYphWamiXI,405
@@ -104,10 +104,10 @@ torchx/util/types.py,sha256=E9dxAWQnsJkIDuHtg-poeOJ4etucSI_xP_Z5kNJX8uI,9229
 torchx/workspace/__init__.py,sha256=FqN8AN4VhR1C_SBY10MggQvNZmyanbbuPuE-JCjkyUY,798
 torchx/workspace/api.py,sha256=UESQ4qgxXjsb6Y1wP9OGv2ixaFgaTs3SqghmNuOJIZM,10235
 torchx/workspace/dir_workspace.py,sha256=npNW_IjUZm_yS5r-8hrRkH46ndDd9a_eApT64m1S1T4,2268
-torchx/workspace/docker_workspace.py,sha256=PFu2KQNVC-0p2aKJ-W_BKA9ZOmXdCY2ABEkCExp3udQ,10269
-torchx_nightly-2026.1.11.dist-info/licenses/LICENSE,sha256=WVHfXhFC0Ia8LTKt_nJVYobdqTJVg_4J3Crrfm2A8KQ,1721
-torchx_nightly-2026.1.11.dist-info/METADATA,sha256=VzSwxPN0aaQV3U3gNuMZMvhXiVRwO3W51DLXH1jaEr0,5323
-torchx_nightly-2026.1.11.dist-info/WHEEL,sha256=SmOxYU7pzNKBqASvQJ7DjX3XGUF92lrGhMb3R6_iiqI,91
-torchx_nightly-2026.1.11.dist-info/entry_points.txt,sha256=T328AMXeKI3JZnnxfkEew2ZcMN1oQDtkXjMz7lkV-P4,169
-torchx_nightly-2026.1.11.dist-info/top_level.txt,sha256=pxew3bc2gsiViS0zADs0jb6kC5v8o_Yy_85fhHj_J1A,7
-torchx_nightly-2026.1.11.dist-info/RECORD,,
+torchx/workspace/docker_workspace.py,sha256=EkiveycTlCYPzrkkoqL2EXNFZSUc3015RgTQY-7a3iU,10268
+torchx_nightly-2026.1.22.dist-info/licenses/LICENSE,sha256=WVHfXhFC0Ia8LTKt_nJVYobdqTJVg_4J3Crrfm2A8KQ,1721
+torchx_nightly-2026.1.22.dist-info/METADATA,sha256=D169Ar4bVrkBHjTLDbenoqIyqBEiqxoDtDc59YXl4N8,5323
+torchx_nightly-2026.1.22.dist-info/WHEEL,sha256=SmOxYU7pzNKBqASvQJ7DjX3XGUF92lrGhMb3R6_iiqI,91
+torchx_nightly-2026.1.22.dist-info/entry_points.txt,sha256=T328AMXeKI3JZnnxfkEew2ZcMN1oQDtkXjMz7lkV-P4,169
+torchx_nightly-2026.1.22.dist-info/top_level.txt,sha256=pxew3bc2gsiViS0zADs0jb6kC5v8o_Yy_85fhHj_J1A,7
+torchx_nightly-2026.1.22.dist-info/RECORD,,