diff --git a/project/backend/coordinator/controller/kpi_setting_controller.py b/project/backend/coordinator/controller/kpi_setting_controller.py index ff037a0..c19a81e 100644 --- a/project/backend/coordinator/controller/kpi_setting_controller.py +++ b/project/backend/coordinator/controller/kpi_setting_controller.py @@ -59,7 +59,7 @@ def create_kpi_setting(): position=data["position"], active=data["active"], examples=data.get("examples", []), - is_trained=False + is_trained=False, ) db.session.add(new_kpi_setting) @@ -104,6 +104,9 @@ def update_kpi_setting(id): if "examples" in data: kpi_setting.examples = data["examples"] + if "is_trained" in data: + kpi_setting.is_trained = data["is_trained"] + db.session.commit() return jsonify(kpi_setting.to_dict()), 200 diff --git a/project/backend/coordinator/controller/spacy_controller.py b/project/backend/coordinator/controller/spacy_controller.py index eeb207c..330e9e5 100644 --- a/project/backend/coordinator/controller/spacy_controller.py +++ b/project/backend/coordinator/controller/spacy_controller.py @@ -131,7 +131,15 @@ def update_training_status(): db.session.commit() except Exception as e: db.session.rollback() - return jsonify({"error": "is_trained konnte nicht aktualisiert werden", "details": str(e)}), 500 + return ( + jsonify( + { + "error": "is_trained konnte nicht aktualisiert werden", + "details": str(e), + } + ), + 500, + ) return jsonify({"status": "success", "running": current_training_status["running"]}) diff --git a/project/backend/coordinator/model/kpi_setting_model.py b/project/backend/coordinator/model/kpi_setting_model.py index 37936b2..b72444a 100644 --- a/project/backend/coordinator/model/kpi_setting_model.py +++ b/project/backend/coordinator/model/kpi_setting_model.py @@ -29,7 +29,6 @@ class KPISettingModel(db.Model): examples: Mapped[list] = mapped_column(JSONB, default=[]) is_trained: Mapped[bool] = mapped_column(default=False) - def to_dict(self): return OrderedDict( [ @@ -44,7 +43,9 @@ class KPISettingModel(db.Model): ] ) - def __init__(self, name, mandatory, type, position, active, examples=None, is_trained=False): + def __init__( + self, name, mandatory, type, position, active, examples=None, is_trained=False + ): self.name = name self.mandatory = mandatory self.type = type diff --git a/project/backend/coordinator/model/seed_data.py b/project/backend/coordinator/model/seed_data.py index 7ca5396..da0a64b 100644 --- a/project/backend/coordinator/model/seed_data.py +++ b/project/backend/coordinator/model/seed_data.py @@ -289,4 +289,3 @@ def seed_default_kpi_settings(): db.session.rollback() print(f"Fehler beim Hinzufügen der Standard KPI Settings: {e}") raise -