diff --git a/analyzer/ml/tests/test_alert_notifier.py b/analyzer/ml/tests/test_alert_notifier.py index d78d49d..93de971 100644 --- a/analyzer/ml/tests/test_alert_notifier.py +++ b/analyzer/ml/tests/test_alert_notifier.py @@ -189,13 +189,14 @@ def setUp(self): cache.clear() def test_evaluation_sends_one_email_per_new_alert(self): + from django.utils import timezone as dj_timezone + from analyzer.ml.monitoring import alert_evaluator from analyzer.ml.monitoring.retraining_system import ( RetrainingTrigger, TriggerReason, TriggerUrgency, ) - from django.utils import timezone as dj_timezone triggers = [ RetrainingTrigger( diff --git a/analyzer/urls.py b/analyzer/urls.py index e7a481f..94d59df 100644 --- a/analyzer/urls.py +++ b/analyzer/urls.py @@ -9,7 +9,6 @@ # ML Dashboard views (separate module) from .ml import dashboard_views -from .views import ml_alert_views # Import from modular views package from .views import ( # Authentication views; Query grading views; Comparison views; Batch analysis views; History and feedback views; Upload views; Database introspection views; Async processing views; API views; Saved connection views @@ -39,6 +38,7 @@ index, login_view, logout_view, + ml_alert_views, password_change, password_reset_confirm, password_reset_request, diff --git a/analyzer/views/ml_alert_views.py b/analyzer/views/ml_alert_views.py index 8d916a4..ebcdaf7 100644 --- a/analyzer/views/ml_alert_views.py +++ b/analyzer/views/ml_alert_views.py @@ -20,8 +20,11 @@ from django.utils import timezone from django.views.decorators.http import require_POST -from analyzer.ml.monitoring.rollback import can_rollback -from analyzer.ml.monitoring.rollback import RollbackError, perform_rollback +from analyzer.ml.monitoring.rollback import ( + RollbackError, + can_rollback, + perform_rollback, +) from analyzer.models import MLAlert, MLModel logger = logging.getLogger(__name__)