diff --git a/.github/workflows/pr.yaml b/.github/workflows/pr.yaml index d4f67ee..3d730e0 100644 --- a/.github/workflows/pr.yaml +++ b/.github/workflows/pr.yaml @@ -129,7 +129,7 @@ jobs: - name: Set fail level for pull request if: ${{ github.event_name == 'pull_request' }} run: - echo "FAIL_LEVEL=warn" >> "$GITHUB_ENV" + echo "FAIL_LEVEL=error" >> "$GITHUB_ENV" - name: Set fail level for merge if: ${{ github.event_name != 'pull_request' }} run: diff --git a/tools/base_model_trainer.py b/tools/base_model_trainer.py index 1b1104c..2f3a689 100644 --- a/tools/base_model_trainer.py +++ b/tools/base_model_trainer.py @@ -1,6 +1,6 @@ -import os -import logging import base64 +import logging +import os import pandas as pd diff --git a/tools/dashboard.py b/tools/dashboard.py index 328b36f..7d1fad3 100644 --- a/tools/dashboard.py +++ b/tools/dashboard.py @@ -1,5 +1,5 @@ -from typing import Any, Dict, Optional import logging +from typing import Any, Dict, Optional from pycaret.utils.generic import get_label_encoder diff --git a/tools/pycaret_classification.py b/tools/pycaret_classification.py index 9efde46..099ef1b 100644 --- a/tools/pycaret_classification.py +++ b/tools/pycaret_classification.py @@ -1,7 +1,9 @@ import logging from base_model_trainer import BaseModelTrainer + from pycaret.classification import ClassificationExperiment + from dashboard import generate_classifier_explainer_dashboard LOG = logging.getLogger(__name__) diff --git a/tools/pycaret_regression.py b/tools/pycaret_regression.py index 404d196..282ab8d 100644 --- a/tools/pycaret_regression.py +++ b/tools/pycaret_regression.py @@ -1,7 +1,9 @@ import logging from base_model_trainer import BaseModelTrainer + from pycaret.regression import RegressionExperiment + from dashboard import generate_regression_explainer_dashboard LOG = logging.getLogger(__name__) diff --git a/tools/pycaret_train.py b/tools/pycaret_train.py index ef667bc..1f15c8f 100644 --- a/tools/pycaret_train.py +++ b/tools/pycaret_train.py @@ -2,6 +2,7 @@ import logging from pycaret_classification import ClassificationModelTrainer + from pycaret_regression import RegressionModelTrainer logging.basicConfig(level=logging.DEBUG)