feat: finish working bfs example

1 files changed, 6 insertions, 2 deletions

diff --git a/src/research.py b/src/research.py
index 2a59c49..8c46877 100644
--- a/src/research.py
+++ b/src/research.py
@@ -34,7 +34,7 @@ class Research:
 
             if iteration % 10 == 0 or iteration == 1:
                 if preview or save_plot:
-                    self.plotter.plot(self.model, normalize=False, density=1)
+                    self.plotter.plot(self.model, normalize=True)
                 if preview:
                     self.plotter.show()
                 if save_plot:
@@ -45,10 +45,14 @@ class Research:
                     self.model.save(self.model_path)
 
         self.model.save(self.solution_path)
+
+        self.plotter.plot(self.model, streamplot=True)
+        self.plotter.save(os.path.join(self.path, f'streamplot.png'))
+
         self.inspect()
 
     def inspect(self):
-        self.plotter.plot(self.model, density=1)
+        self.plotter.plot(self.model, streamplot=True)
         while True:
             self.plotter.show()