@@ -419,12 +419,7 @@ def dryrun(
419
419
sched = self ._scheduler (scheduler )
420
420
resolved_cfg = sched .run_opts ().resolve (cfg )
421
421
422
- # early validation before build workspace
423
- with log_event (
424
- "pre_build_validate" ,
425
- scheduler ,
426
- ):
427
- sched ._pre_build_validate (app , scheduler , resolved_cfg )
422
+ sched ._pre_build_validate (app , scheduler , resolved_cfg )
428
423
429
424
if workspace and isinstance (sched , WorkspaceMixin ):
430
425
role = app .roles [0 ]
@@ -434,13 +429,7 @@ def dryrun(
434
429
logger .info (
435
430
'To disable workspaces pass: --workspace="" from CLI or workspace=None programmatically.'
436
431
)
437
- with log_event (
438
- "build_workspace_and_update_role" ,
439
- scheduler ,
440
- ) as ctx :
441
- sched .build_workspace_and_update_role (role , workspace , resolved_cfg )
442
- ctx ._torchx_event .app_image = role .image
443
- ctx ._torchx_event .workspace = workspace
432
+ sched .build_workspace_and_update_role (role , workspace , resolved_cfg )
444
433
445
434
if old_img != role .image :
446
435
logger .info (
@@ -453,11 +442,7 @@ def dryrun(
453
442
" Either a patch was built or no changes to workspace was detected."
454
443
)
455
444
456
- with log_event (
457
- "validate" ,
458
- scheduler ,
459
- ):
460
- sched ._validate (app , scheduler , resolved_cfg )
445
+ sched ._validate (app , scheduler , resolved_cfg )
461
446
dryrun_info = sched .submit_dryrun (app , resolved_cfg )
462
447
dryrun_info ._scheduler = scheduler
463
448
return dryrun_info
0 commit comments