From 213e24cafd4da6aba34ce68ac2ab83be132a8b81 Mon Sep 17 00:00:00 2001 From: Stephanie Wang Date: Wed, 31 Aug 2022 20:43:58 -0400 Subject: [PATCH] [tests] Remove unnecessary sleep time from pipelined ingest tests #28182 Signed-off-by: Stephanie Wang --- release/nightly_tests/dataset/pipelined_training.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/release/nightly_tests/dataset/pipelined_training.py b/release/nightly_tests/dataset/pipelined_training.py index 83d0f5804..69c24014d 100644 --- a/release/nightly_tests/dataset/pipelined_training.py +++ b/release/nightly_tests/dataset/pipelined_training.py @@ -44,7 +44,6 @@ parser.add_argument( help=("how many batches to wait before logging training " "status"), ) parser.add_argument("--num-workers", type=int, default=16) -parser.add_argument("--mock-train-step-time", type=float, default=1.0) parser.add_argument("--num-files", type=int, default=30) parser.add_argument("--num-windows", type=int, default=1) parser.add_argument("--manual-windows", type=bool, default=False) @@ -137,7 +136,6 @@ def train_main(args, splits): f"Processing batch {batch_idx} in epoch {epoch} on worker " f"{rank}." ) - time.sleep(args.mock_train_step_time) loss = loss_function(batch_pred, target, delta=60) loss.mean().backward() for opt in optimizers: