From f051e070351d0618c304015b20401b745bdd10c2 Mon Sep 17 00:00:00 2001
From: Till Raab <till.raab@uni-tuebingen.de>
Date: Fri, 20 Oct 2023 11:35:20 +0200
Subject: [PATCH] bg

---
 data/generate_dataset.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/data/generate_dataset.py b/data/generate_dataset.py
index 67ab95e..f106237 100644
--- a/data/generate_dataset.py
+++ b/data/generate_dataset.py
@@ -125,9 +125,7 @@ def main(args):
                                            (rise_size[id_idx] >= 10)], dtype=int)
                 ax.plot(times_v[rise_idx_oi], fish_freq[id_idx][rise_idx_oi], 'o', color='tab:red')
 
-            plt.show(block=False)
-            time.sleep(10)
-            plt.close()
+            plt.show()
 
 
 if __name__ == '__main__':