blub

python_package/src/onlinemid/dataplotters/gain_stage.py

@@ -12,6 +12,9 @@ class GainStagePlotter():
     def __init__(self):
         self.onlineImagePlotter = OnlineImagePlotter(figsize=(15,10))
 
+    def close(self):
+        self.onlineImagePlotter.close()
+        
     def plot(self, train_id, gainstage_data):
 
         if gainstage_data.shape[1] != 3: