File tree Expand file tree Collapse file tree 1 file changed +7
-10
lines changed Expand file tree Collapse file tree 1 file changed +7
-10
lines changed Original file line number Diff line number Diff line change @@ -470,20 +470,17 @@ def test_dump_and_load_all_registered_schedulers(self) -> None:
470
470
sfile = StringIO ()
471
471
dump (sfile )
472
472
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 ()
482
474
483
475
for sched_name , sched in scheduler_factories .items ():
484
476
sfile .seek (0 ) # reset the file pos
485
477
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
+
487
484
for opt_name , _ in sched ("test" ).run_opts ():
488
485
self .assertTrue (
489
486
opt_name in cfg ,
You can’t perform that action at this time.
0 commit comments