fix: correcao de importacao

heltonricardo · Dec 20, 2023 · 786443d · 786443d
1 parent 73c07ab
commit 786443d
Show file tree

Hide file tree

Showing 4 changed files with 3 additions and 3 deletions.
diff --git 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 → src/app/main.py b/app/main.py → src/app/main.py
diff --git a/test/test_train.py → src/test/test_train.py b/test/test_train.py → 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 → src/train/train.py b/train.py → src/train/train.py