lint

Author: wwwjn

.github/workflows/integration_test_8gpu_models.yaml

@@ -54,4 +54,4 @@ jobs:
         python -m tests.integration_tests.run_tests --test_suite models artifacts-to-be-uploaded --ngpu 8
         python -m tests.integration_tests.flux artifacts-to-be-uploaded/flux --ngpu 8
         rm -rf artifacts-to-be-uploaded/*/checkpoint
-        rm -rf artifacts-to-be-uploaded/flux/test_generate/inference_results/
+        rm -rf artifacts-to-be-uploaded/flux/*/inference_results/

tests/integration_tests/flux.py

@@ -29,7 +29,7 @@ def build_flux_test_list() -> list[OverrideDefinitions]:
                     "--validation.enable",
                     "--checkpoint.enable",
                 ],
-                []
+                [],
             ],
             "HSDP+CP+Validation+Inference",
             "hsdp+cp+validation+inference",