fix module load order

.gitlab-ci.yml

@@ -30,15 +30,15 @@ prepare_data:
     SLURM_PARAMTIME: "-t 10:00"
   stage: prepare_data
   before_script:
-    - python -m pip install poetry
-    - poetry config virtualenvs.create false
-    - poetry install
-  script:
     - module load GCCcore/.13.3.0
     - module load Python/3.12.3
     - module load GCC/12.3.0
     - module load OpenMPI/4.1.5
     - module load PyTorch/2.1.2-CUDA-12.1.1
+    - python -m pip install poetry
+    - poetry config virtualenvs.create false
+    - poetry install
+  script:
     - echo "I would be preparing the data..."
 
 test_hpc:
@@ -51,11 +51,12 @@ test_hpc:
   variables:
     SLURM_PARAM_CPUS: "--cpus-per-task 1"
     SLURM_PARAMTIME: "-t 10:00"
-  script:
+  before_script:
     - module load GCCcore/.13.3.0
     - module load Python/3.12.3
     - module load GCC/12.3.0
     - module load OpenMPI/4.1.5
     - module load PyTorch/2.1.2-CUDA-12.1.1
+  script:
     - python -m pip install --user python-dotenv matplotlib tqdm torchvision==0.16
     - python main.py --debug