diff --git a/.github/workflows/train.yml b/.github/workflows/train.yml
index be8cfbc..b40d4cb 100644
--- a/.github/workflows/train.yml
+++ b/.github/workflows/train.yml
@@ -2,12 +2,12 @@ name: Test Training Code with gRPC
 
 on:
   # used for debugging purposes
-  workflow_dispatch:
+  # workflow_dispatch:
   push:
     branches:
       # run test on push to main only
-      # - main
-      - "*"
+      - main
+      # - "*"
   pull_request:
     branches:
       - main