ruff format

Signed-off-by: Sukriti-Sharma4 <sukriti.sharma4@ibm.com>

Author: Ssukriti

aiu_fms_testing_utils/scripts/drive_paged_programs.py

@@ -398,9 +398,9 @@ def __load_validation_info(
 # input_ids, extra_kwargs, sample_key = __prepare_inputs(2, max_tkv, tokenizer)
 pad_multiple = 64
 if args.prefill_chunk_size > 0:
-    assert (
-        args.prefill_chunk_size % 64 == 0
-    ), "Chunk size must be a multiple of the page size"
+    assert args.prefill_chunk_size % 64 == 0, (
+        "Chunk size must be a multiple of the page size"
+    )
     pad_multiple = args.prefill_chunk_size
 prompt_list = [torch.arange(0, pad_multiple, dtype=torch.int64)]
 # matching vllm warmup to pad to 2 on fp8, and no pad for fp16