diff --git a/.github/workflows/run_pytest.yml b/.github/workflows/run_pytest.yml deleted file mode 100644 index 30fedb6..0000000 --- a/.github/workflows/run_pytest.yml +++ /dev/null @@ -1,34 +0,0 @@ -name: Run Pytest - -on: - push: - branches: "main" - pull_request: - branches: "main" - workflow_dispatch: - -jobs: - test: - runs-on: ubuntu-latest - - steps: - - name: Checkout repository - uses: actions/checkout@v3 - - - name: Set up Python - uses: actions/setup-python@v4 - with: - python-version: "3.11" - - - name: Install Poetry - run: | - curl -sSL https://install.python-poetry.org | python3 - - - - name: Install dependencies - run: | - poetry install - - - name: Run tests - run: | - set -e - poetry run pytest tests/ diff --git a/views_stepshifter/manager/stepshifter_manager.py b/views_stepshifter/manager/stepshifter_manager.py index cf962a6..4c076e7 100644 --- a/views_stepshifter/manager/stepshifter_manager.py +++ b/views_stepshifter/manager/stepshifter_manager.py @@ -1,5 +1,4 @@ -from views_pipeline_core.managers.model_manager import ModelManager -from views_pipeline_core.managers.path_manager import ModelPath +from views_pipeline_core.managers.model import ModelPathManager, ModelManager from views_pipeline_core.models.outputs import generate_output_dict from views_pipeline_core.files.utils import read_log_file, create_log_file from views_pipeline_core.wandb.utils import add_wandb_monthly_metrics, generate_wandb_log_dict, log_wandb_log_dict @@ -23,7 +22,7 @@ class StepshifterManager(ModelManager): - def __init__(self, model_path: ModelPath) -> None: + def __init__(self, model_path: ModelPathManager) -> None: super().__init__(model_path) self._is_hurdle = self._config_meta["algorithm"] == "HurdleModel" diff --git a/views_stepshifter/models/hurdle_model.py b/views_stepshifter/models/hurdle_model.py index 15fe99f..4e641d8 100644 --- a/views_stepshifter/models/hurdle_model.py +++ b/views_stepshifter/models/hurdle_model.py @@ -1,4 +1,4 @@ -from views_pipeline_core.managers.model_manager import ModelManager +from views_pipeline_core.managers.model import ModelManager from views_stepshifter.models.stepshifter import StepshifterModel from views_stepshifter.models.validation import views_validate from views_forecasts.extensions import * diff --git a/views_stepshifter/models/stepshifter.py b/views_stepshifter/models/stepshifter.py index 04fd6c7..a2fa50d 100644 --- a/views_stepshifter/models/stepshifter.py +++ b/views_stepshifter/models/stepshifter.py @@ -6,7 +6,7 @@ from typing import List, Dict from views_forecasts.extensions import * from views_stepshifter.models.validation import views_validate -from views_pipeline_core.managers.model_manager import ModelManager +from views_pipeline_core.managers.model import ModelManager logger = logging.getLogger(__name__)