From 786443dbef8982d8893de82f7896e4b26d89a1c4 Mon Sep 17 00:00:00 2001
From: Helton Ricardo <50843386+heltonricardo@users.noreply.github.com>
Date: Wed, 20 Dec 2023 20:15:42 -0300
Subject: [PATCH] fix: correcao de importacao

---
 .github/workflows/pipeline.yml   | 4 ++--
 {app => src/app}/main.py         | 0
 {test => src/test}/test_train.py | 2 +-
 train.py => src/train/train.py   | 0
 4 files changed, 3 insertions(+), 3 deletions(-)
 rename {app => src/app}/main.py (100%)
 rename {test => src/test}/test_train.py (96%)
 rename train.py => src/train/train.py (100%)

diff --git a/.github/workflows/pipeline.yml b/.github/workflows/pipeline.yml
index d8e3f06..eb0fc02 100644
--- a/.github/workflows/pipeline.yml
+++ b/.github/workflows/pipeline.yml
@@ -31,7 +31,7 @@ jobs:
           if [ -f requirements.txt ]; then pip install -r requirements.txt; fi
 
       - name: Test files
-        run: pytest test/test_train.py
+        run: pytest src/test/test_train.py
 
   check_python_code_style:
     name: Check Python code style
@@ -74,7 +74,7 @@ jobs:
           if [ -f requirements.txt ]; then pip install -r requirements.txt; fi
 
       - name: Train Model
-        run: python train.py
+        run: python src/train/train.py
 
   # build_image:
   #   name: Build image
diff --git a/app/main.py b/src/app/main.py
similarity index 100%
rename from app/main.py
rename to src/app/main.py
diff --git a/test/test_train.py b/src/test/test_train.py
similarity index 96%
rename from test/test_train.py
rename to src/test/test_train.py
index 41b8b8a..b23e27e 100644
--- a/test/test_train.py
+++ b/src/test/test_train.py
@@ -2,7 +2,7 @@
 import pytest
 from tensorflow.keras.models import Sequential
 
-from train import read_data, create_model, train_model
+from ..train.train import read_data, create_model, train_model
 
 
 @pytest.fixture
diff --git a/train.py b/src/train/train.py
similarity index 100%
rename from train.py
rename to src/train/train.py