Skip to content

Commit 2d59334

Browse files
authored
(6/n - non-xlformers conda-on-mast mvp)(torchx/runner) Create //torchx/runner:lib_core with no plugin bundling
Differential Revision: D77400710 Pull Request resolved: #1084
1 parent 3dcab69 commit 2d59334

File tree

1 file changed

+7
-10
lines changed

1 file changed

+7
-10
lines changed

torchx/runner/test/config_test.py

Lines changed: 7 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -470,20 +470,17 @@ def test_dump_and_load_all_registered_schedulers(self) -> None:
470470
sfile = StringIO()
471471
dump(sfile)
472472

473-
scheduler_factories = {
474-
**get_scheduler_factories(),
475-
**(
476-
get_scheduler_factories(
477-
group="torchx.schedulers.orchestrator", skip_defaults=True
478-
)
479-
or {}
480-
),
481-
}
473+
scheduler_factories = get_scheduler_factories()
482474

483475
for sched_name, sched in scheduler_factories.items():
484476
sfile.seek(0) # reset the file pos
485477
cfg = {}
486-
load(scheduler=sched_name, f=sfile, cfg=cfg)
478+
try:
479+
load(scheduler=sched_name, f=sfile, cfg=cfg)
480+
except ModuleNotFoundError:
481+
# just test the ones that have been installed
482+
continue
483+
487484
for opt_name, _ in sched("test").run_opts():
488485
self.assertTrue(
489486
opt_name in cfg,

0 commit comments

Comments
 (0)