From 9a041824c5e66c951a704f817da158d58985a104 Mon Sep 17 00:00:00 2001
From: Rudio <rudio.fida_cyrille@etu.sorbonne-universite.fr>
Date: Fri, 8 Dec 2023 21:24:02 +0100
Subject: [PATCH] fixing tests

---
 tests/test_train.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/tests/test_train.py b/tests/test_train.py
index edf0cbf..55c4175 100644
--- 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