From a58d81159c03b76d30e6d58e721fa3658eaa805c Mon Sep 17 00:00:00 2001
From: S1ro1 <matej.sirovatka@gmail.com>
Date: Sat, 9 Nov 2024 13:18:08 +0100
Subject: [PATCH] Fix: move ci debug up

---
 .github/workflows/train_workflow.yml | 15 ++++++++-------
 1 file changed, 8 insertions(+), 7 deletions(-)

diff --git a/.github/workflows/train_workflow.yml b/.github/workflows/train_workflow.yml
index c42c554..5255892 100644
--- a/.github/workflows/train_workflow.yml
+++ b/.github/workflows/train_workflow.yml
@@ -24,13 +24,6 @@ jobs:
           pip install numpy
           # Add other Python dependencies as needed
 
-      - name: Install CUDA dependencies
-        if: inputs.script_type == 'cu'
-        run: |
-          sudo apt-get update
-          sudo apt-get install -y nvidia-cuda-toolkit
-          nvcc --version
-        
       - name: Debug Step
         if: inputs.script_type == 'cu'
         run: |
@@ -40,6 +33,14 @@ jobs:
           EOL
           cat train.${{ inputs.script_type }}
 
+      - name: Install CUDA dependencies
+        if: inputs.script_type == 'cu'
+        run: |
+          sudo apt-get update
+          sudo apt-get install -y nvidia-cuda-toolkit
+          nvcc --version
+        
+
       - name: Create training script
         if : inputs.script_type == 'py' # TODO: remove later
         run: |