From dc5131136e40d3764afc4bb1d9ed1de950d25e44 Mon Sep 17 00:00:00 2001
From: PawelDopierala <pawel.dopierala4@gmail.com>
Date: Thu, 6 Jun 2024 02:18:13 +0200
Subject: [PATCH] fix github workflow

---
 .github/workflows/main.yml | 30 ++++++++++++++++++++++++++++--
 1 file changed, 28 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
index a6c0e17..c4f4315 100644
--- a/.github/workflows/main.yml
+++ b/.github/workflows/main.yml
@@ -17,7 +17,7 @@ on:
         default: 32
 
 jobs:
-  train_and_evaluate:
+  install_dependencies:
     runs-on: ubuntu-latest
 
     steps:
@@ -34,8 +34,34 @@ jobs:
         python -m pip install --upgrade pip
         pip install pandas scikit-learn tensorflow matplotlib mlflow
 
+  train_model:
+    needs: install_dependencies
+    runs-on: ubuntu-latest
+
+    steps:
+    - name: Checkout repository
+      uses: actions/checkout@v2
+
+    - name: Set up Python
+      uses: actions/setup-python@v2
+      with:
+        python-version: '3.8'
+
     - name: Train Model
       run: python ./github_project/create_model.py ${{ github.event.inputs.epochs }} ${{ github.event.inputs.learning_rate }} ${{ github.event.inputs.batch_size }}
 
+  evaluate_model:
+    needs: install_dependencies
+    runs-on: ubuntu-latest
+
+    steps:
+    - name: Checkout repository
+      uses: actions/checkout@v2
+
+    - name: Set up Python
+      uses: actions/setup-python@v2
+      with:
+        python-version: '3.8'
+
     - name: Evaluate Model
-      run: python ./github_project/evaluate.py ${{ github.run_number }}
\ No newline at end of file
+      run: python ./github_project/evaluate.py ${{ github.run_number }}