From 031b2798addadd1652db7cfba50eacc1079245cf Mon Sep 17 00:00:00 2001
From: Sayak Paul <spsayakpaul@gmail.com>
Date: Wed, 30 Sep 2026 10:47:09 +0530
Subject: [PATCH] spit output to stderr when extracting GPU tests for PRs
 (#14847)

* update

* don't use xdist during gpu tests in PRs.

* Revert "update"

This reverts commit 0906d05c8b55aeeef52a1a2a114d8066a19a05d9.

* fix(ci): harden GitHub Actions workflows (#14847) (#14849)

fix(ci): harden workflow files flagged on #14847

Co-authored-by: hf-security-analysis[bot] <265538906+hf-security-analysis[bot]@users.noreply.github.com>

* redirect to stderr

---------

Co-authored-by: DN6 <dhruv.nair@gmail.com>
Co-authored-by: hf-security-analysis[bot] <265538906+hf-security-analysis[bot]@users.noreply.github.com>
---
 utils/extract_tests_from_mixin.py | 52 ++++++++++++++++---------------
 1 file changed, 27 insertions(+), 25 deletions(-)

diff --git a/utils/extract_tests_from_mixin.py b/utils/extract_tests_from_mixin.py
index 04b157ff50..b52ec40f64 100644
--- a/utils/extract_tests_from_mixin.py
+++ b/utils/extract_tests_from_mixin.py
@@ -1,6 +1,7 @@
 import argparse
 import inspect
 import sys
+from contextlib import redirect_stdout
 from pathlib import Path
 from typing import List, Type
 
@@ -41,35 +42,36 @@ def generate_pattern_for_mixins(mixin_classes: List[Type]) -> str:
 
 
 if __name__ == "__main__":
-    mixin_classes = []
-    if args.type == "pipeline":
-        from tests.pipelines.test_pipelines_common import PipelineTesterMixin
+    with redirect_stdout(sys.stderr):
+        mixin_classes = []
+        if args.type == "pipeline":
+            from tests.pipelines.test_pipelines_common import PipelineTesterMixin
 
-        mixin_classes = [PipelineTesterMixin]
+            mixin_classes = [PipelineTesterMixin]
 
-    elif args.type == "models":
-        # The model tester suite is split across several mixins under `tests/models/testing_utils`,
-        # so aggregate their test methods to reconstruct the full coverage.
-        from tests.models.testing_utils import (
-            AttentionTesterMixin,
-            LoraTesterMixin,
-            MemoryTesterMixin,
-            ModelTesterMixin,
-            TrainingTesterMixin,
-        )
+        elif args.type == "models":
+            # The model tester suite is split across several mixins under `tests/models/testing_utils`,
+            # so aggregate their test methods to reconstruct the full coverage.
+            from tests.models.testing_utils import (
+                AttentionTesterMixin,
+                LoraTesterMixin,
+                MemoryTesterMixin,
+                ModelTesterMixin,
+                TrainingTesterMixin,
+            )
 
-        mixin_classes = [
-            ModelTesterMixin,
-            MemoryTesterMixin,
-            TrainingTesterMixin,
-            AttentionTesterMixin,
-            LoraTesterMixin,
-        ]
+            mixin_classes = [
+                ModelTesterMixin,
+                MemoryTesterMixin,
+                TrainingTesterMixin,
+                AttentionTesterMixin,
+                LoraTesterMixin,
+            ]
 
-    elif args.type == "lora":
-        from tests.lora.utils import PeftLoraLoaderMixinTests
+        elif args.type == "lora":
+            from tests.lora.utils import PeftLoraLoaderMixinTests
 
-        mixin_classes = [PeftLoraLoaderMixinTests]
+            mixin_classes = [PeftLoraLoaderMixinTests]
 
-    pattern = generate_pattern_for_mixins(mixin_classes)
+        pattern = generate_pattern_for_mixins(mixin_classes)
     print(pattern)