Update test_llm_api_pytorch.py

Signed-off-by: Yi Zhang <[email protected]>

Author: yizhang-nv

tests/integration/defs/accuracy/test_llm_api_pytorch.py

@@ -871,8 +871,10 @@ def test_nvfp4_4gpus(self, fp8kv, attention_dp, cuda_graph,
             pytest.skip("https://nvbugs/5252559")
         if torch_compile and pp_size > 1:
             pytest.skip("PP with torch.compile is not supported yet.")
-        if not attention_dp and (tp_size > 1 or ep_size > 1) and moe_backend == "CUTLASS":
+        if not attention_dp and (tp_size > 1
+                                 or ep_size > 1) and moe_backend == "CUTLASS":
             pytest.skip("https://nvbugs/5336321")
+
         kv_cache_config = KvCacheConfig(free_gpu_memory_fraction=0.9)
         torch_compile_config = TorchCompileConfig(
             enable_fullgraph=True,