diff --git a/torchx/runner/api.py b/torchx/runner/api.py index 4a03a8aad..c9a6faeb1 100644 --- a/torchx/runner/api.py +++ b/torchx/runner/api.py @@ -433,7 +433,11 @@ def dryrun( " Either a patch was built or no changes to workspace was detected." ) - sched._validate(app, scheduler) + with log_event( + "validate", + scheduler, + ): + sched._validate(app, scheduler) dryrun_info = sched.submit_dryrun(app, resolved_cfg) dryrun_info._scheduler = scheduler return dryrun_info