fixing tests

MLOps-essi-upc · Dec 8, 2023 · 9a04182 · 9a04182
1 parent e6d87c8
commit 9a04182
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/tests/test_train.py b/tests/test_train.py
@@ -1,4 +1,13 @@
 import pytest
+import os
+import sys
+
+# Get the parent directory
+parent_dir = os.path.dirname(os.path.realpath(__file__))
+
+# Add the parent directory to sys.path
+sys.path.append(parent_dir)
+
 from src.models.predict_model import predict,SentiBites
 from src.models.train_model import pre_processing, train
 import datasets