send status message on done training

fiebrink1 · Jan 10, 2019 · 3ece354 · 3ece354
1 parent 7f14dfc
commit 3ece354
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/wekimini/SupervisedLearningManager.java b/src/wekimini/SupervisedLearningManager.java
@@ -403,6 +403,10 @@ public LearningState getLearningState() {
     private void setLearningState(LearningState learningState) {
         LearningState oldLearningState = this.learningState;
         this.learningState = learningState;
+        if(this.learningState == SupervisedLearningManager.LearningState.DONE_TRAINING)
+        {
+            w.getOutputManager().sendMessage("/wekinator/status/done-training");
+        }
         updateAbleToRun();
         propertyChangeSupport.firePropertyChange(PROP_LEARNINGSTATE, oldLearningState, learningState);
     }