From 3764ecbd59828a70b29c59b6b17b5b74f32d54b4 Mon Sep 17 00:00:00 2001 From: knikolaou Date: Mon, 13 May 2024 17:21:33 +0200 Subject: [PATCH] fix pytest file search --- CI/unit_tests/measurements/test_base_measurement.py | 2 +- CI/unit_tests/neural_state/test_neural_state.py | 2 +- CI/unit_tests/neural_state/test_neural_state_creator.py | 2 +- papyrus/__init__.py | 3 ++- pyproject.toml | 3 +++ 5 files changed, 8 insertions(+), 4 deletions(-) diff --git a/CI/unit_tests/measurements/test_base_measurement.py b/CI/unit_tests/measurements/test_base_measurement.py index 48cbc07..1004404 100644 --- a/CI/unit_tests/measurements/test_base_measurement.py +++ b/CI/unit_tests/measurements/test_base_measurement.py @@ -27,7 +27,7 @@ import numpy as np import pytest -from papyrus.measurements.base_measurement import BaseMeasurement +from papyrus.measurements import BaseMeasurement class DummyMeasurement(BaseMeasurement): diff --git a/CI/unit_tests/neural_state/test_neural_state.py b/CI/unit_tests/neural_state/test_neural_state.py index 0e45871..4ac5381 100644 --- a/CI/unit_tests/neural_state/test_neural_state.py +++ b/CI/unit_tests/neural_state/test_neural_state.py @@ -21,7 +21,7 @@ ------- """ -from papyrus.neural_state.neural_state import NeuralState +from papyrus.neural_state import NeuralState class TestNeuralState: diff --git a/CI/unit_tests/neural_state/test_neural_state_creator.py b/CI/unit_tests/neural_state/test_neural_state_creator.py index 6fb1803..4c30faf 100644 --- a/CI/unit_tests/neural_state/test_neural_state_creator.py +++ b/CI/unit_tests/neural_state/test_neural_state_creator.py @@ -23,7 +23,7 @@ import numpy as np -from papyrus.neural_state.neural_state_creator import NeuralStateCreator +from papyrus.neural_state import NeuralStateCreator class TestNeuralStateCreator: diff --git a/papyrus/__init__.py b/papyrus/__init__.py index 68d7f91..1c18282 100644 --- a/papyrus/__init__.py +++ b/papyrus/__init__.py @@ -22,9 +22,10 @@ papyrus measurements api. """ -from papyrus import measurements, utils +from papyrus import measurements, neural_state, utils __all__ = [ measurements.__name__, utils.__name__, + neural_state.__name__, ] diff --git a/pyproject.toml b/pyproject.toml index f26ac36..bf1881f 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -15,3 +15,6 @@ keywords = ["neural networks", "artificial intelligence"] license = {file = "LICENSE"} version = "0.0.1" + +[tool.pytest.ini_options] +pythonpath = ["papyrus",] \ No newline at end of file