Added legend to saved figures

2022-04-27 15:38:55 +02:00
parent c77e229d9c
commit 40a9bc5d98
1 changed files with 1 additions and 0 deletions
--- a/kalman_filter/display/display_2d.py
+++ b/kalman_filter/display/display_2d.py
@@ -87,6 +87,7 @@ class Displayer:

        for t in range(1, n_steps):
            callback(self, t)
+            self._ax.legend()

            if t in frames:
                self._fig.savefig(prefix + str(t) + "." + ext)