fixed bug with get_optimization_status

rosflight · Apr 12, 2024 · 9d633f3 · 9d633f3
1 parent 6fe8002
commit 9d633f3
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/rosplane_tuning/src/autotune/optimizer.py b/rosplane_tuning/src/autotune/optimizer.py
@@ -66,9 +66,11 @@ def get_optimization_status(self):
         """
         if self.state == OptimizerState.UNINITIALIZED:
             return "Optimizer awaiting first iteration."
+        elif self.state == OptimizerState.FIRST_ITERATION:
+            return "First iteration."
         elif self.state == OptimizerState.OPTIMIZING:
             return "Optimizing."
-        elif self.state == OptimizerState.TERMINATED:
+        else:  # self.state == OptimizerState.TERMINATED:
             return str(self.reason) + " Final gains: " + str(self.x)
 
     def get_next_parameter_set(self, error):