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:
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 ,
You can’t perform that action at this time.
0 commit comments