Skip to content

Commit ff22758

Browse files
authored
fix dump and load behavior to load same set of schedulers
Differential Revision: D67341003 Pull Request resolved: #993
1 parent 90884d7 commit ff22758

File tree

1 file changed

+10
-1
lines changed

1 file changed

+10
-1
lines changed

torchx/runner/config.py

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -249,7 +249,16 @@ def dump(
249249
if schedulers:
250250
scheds = schedulers
251251
else:
252-
scheds = get_scheduler_factories().keys()
252+
scheduler_factories = {
253+
**get_scheduler_factories(),
254+
**(
255+
get_scheduler_factories(
256+
group="torchx.schedulers.orchestrator", skip_defaults=True
257+
)
258+
or {}
259+
),
260+
}
261+
scheds = scheduler_factories.keys()
253262

254263
config = _configparser()
255264
for sched_name in scheds:

0 commit comments

Comments
 (0)