diff --git a/torchx/runner/api.py b/torchx/runner/api.py index 3e194a6c..cd66bd48 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, resolved_cfg) + with log_event( + "validate", + scheduler, + ): + sched._validate(app, scheduler, resolved_cfg) dryrun_info = sched.submit_dryrun(app, resolved_cfg) dryrun_info._scheduler = scheduler return dryrun_info