From f1133629999017dc28b50b30369130464cd0bd84 Mon Sep 17 00:00:00 2001 From: gushob21 Date: Fri, 24 Jan 2025 14:21:16 +0000 Subject: [PATCH] improving data preprocessing script --- .../data-processing/ray/tests/test_datacleaner.py | 2 +- .../data-processing/ray/tests/test_dataloader.py | 2 +- .../data-processing/ray/tests/test_dataprep.py | 2 +- .../data-processing/ray/tests/test_ray_utils.py | 2 +- 4 files changed, 4 insertions(+), 4 deletions(-) diff --git a/use-cases/model-fine-tuning-pipeline/data-processing/ray/tests/test_datacleaner.py b/use-cases/model-fine-tuning-pipeline/data-processing/ray/tests/test_datacleaner.py index 8ef18496..56db570a 100644 --- a/use-cases/model-fine-tuning-pipeline/data-processing/ray/tests/test_datacleaner.py +++ b/use-cases/model-fine-tuning-pipeline/data-processing/ray/tests/test_datacleaner.py @@ -1,6 +1,6 @@ import unittest import pandas as pd -from datapreprocessing.datacleaner import DataPreprocessor +from ..src.datapreprocessing.datacleaner import DataPreprocessor class TestDataCleaner(unittest.TestCase): diff --git a/use-cases/model-fine-tuning-pipeline/data-processing/ray/tests/test_dataloader.py b/use-cases/model-fine-tuning-pipeline/data-processing/ray/tests/test_dataloader.py index bb4cdab2..17db9dd5 100644 --- a/use-cases/model-fine-tuning-pipeline/data-processing/ray/tests/test_dataloader.py +++ b/use-cases/model-fine-tuning-pipeline/data-processing/ray/tests/test_dataloader.py @@ -1,7 +1,7 @@ import unittest import pandas as pd from unittest.mock import patch, Mock -from datapreprocessing.dataloader import DataLoader +from ..src.datapreprocessing.dataloader import DataLoader class TestDataLoader(unittest.TestCase): diff --git a/use-cases/model-fine-tuning-pipeline/data-processing/ray/tests/test_dataprep.py b/use-cases/model-fine-tuning-pipeline/data-processing/ray/tests/test_dataprep.py index ae7fadf2..15c1e142 100644 --- a/use-cases/model-fine-tuning-pipeline/data-processing/ray/tests/test_dataprep.py +++ b/use-cases/model-fine-tuning-pipeline/data-processing/ray/tests/test_dataprep.py @@ -1,6 +1,6 @@ import unittest import pandas as pd -from datapreprocessing.dataprep import DataPrep +from ..src.datapreprocessing.dataprep import DataPrep class TestDataPrep(unittest.TestCase): diff --git a/use-cases/model-fine-tuning-pipeline/data-processing/ray/tests/test_ray_utils.py b/use-cases/model-fine-tuning-pipeline/data-processing/ray/tests/test_ray_utils.py index 3b71541a..b73849f5 100644 --- a/use-cases/model-fine-tuning-pipeline/data-processing/ray/tests/test_ray_utils.py +++ b/use-cases/model-fine-tuning-pipeline/data-processing/ray/tests/test_ray_utils.py @@ -1,7 +1,7 @@ import unittest import pandas as pd from unittest.mock import patch, Mock -from datapreprocessing.ray_utils import RayUtils +from ..src.datapreprocessing.ray_utils import RayUtils class TestRayUtils(unittest.TestCase):