From a6bcff5eb8e016802dc2a026613ccd0ba1ee41e3 Mon Sep 17 00:00:00 2001
From: Daniel Schwartz-Narbonne <danielsn@users.noreply.github.com>
Date: Wed, 11 Dec 2024 10:39:46 -0500
Subject: [PATCH] Update .github/workflows/pytorch_gpu_tests.yml

Co-authored-by: Taegyun Kim <taegyun.kim@datadoghq.com>
---
 .github/workflows/pytorch_gpu_tests.yml | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/pytorch_gpu_tests.yml b/.github/workflows/pytorch_gpu_tests.yml
index bbad1df53e..f715632cd5 100644
--- a/.github/workflows/pytorch_gpu_tests.yml
+++ b/.github/workflows/pytorch_gpu_tests.yml
@@ -1,9 +1,11 @@
 name: Pytorch Unit Tests (with GPU) 
 
 on:
-  pull_request:
+  push:
     branches:
-      - 'main'
+  	  - 'main'
+  	  - 'mq-working-branch**'
+  pull_request:
     paths:
       - 'ddtrace/profiling/collector/pytorch.py'
   workflow_dispatch: