Skip to content

Commit 2199157

Browse files
committed
fix linter
1 parent c43406b commit 2199157

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

MaxText/tests/integration_tests/train_tests.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -322,9 +322,9 @@ def test_gpu_cudnn_flash_jax(self):
322322
cudnn_flash_jax = [ # tests base config on GPU with flash attention"""
323323
None,
324324
os.path.join(PKG_DIR, "configs", "base.yml"),
325-
rf"base_output_directory=gs://runner-maxtext-logs",
325+
"base_output_directory=gs://runner-maxtext-logs",
326326
"run_name=runner_test",
327-
r"dataset_path=gs://maxtext-dataset",
327+
"dataset_path=gs://maxtext-dataset",
328328
"steps=2",
329329
"enable_checkpointing=False",
330330
"enable_goodput_recording=False",

0 commit comments

Comments
 (0)