From 827a2f5b3d021cb90cb4bec3e1f22b1fcf2b90b1 Mon Sep 17 00:00:00 2001 From: Timothy Kuo Date: Fri, 11 Oct 2024 15:20:12 -0400 Subject: [PATCH] run black --- .../huggingface_pytorch/training/resources/scripts/train.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test/sagemaker_tests/huggingface_pytorch/training/resources/scripts/train.py b/test/sagemaker_tests/huggingface_pytorch/training/resources/scripts/train.py index cc278d31c763..c244ed11b1bb 100644 --- a/test/sagemaker_tests/huggingface_pytorch/training/resources/scripts/train.py +++ b/test/sagemaker_tests/huggingface_pytorch/training/resources/scripts/train.py @@ -87,11 +87,11 @@ def run_with_timeout(func, timeout=5): train_dataset = run_with_timeout( train_dataset.map(tokenize, num_proc=1, batched=True, batch_size=len(train_dataset)), - timeout=10 + timeout=10, ) test_dataset = run_with_timeout( test_dataset.map(tokenize, num_proc=1, batched=True, batch_size=len(test_dataset)), - timeout=10 + timeout=10, ) # set format for pytorch