[stat_update] fix bug

leondavi · Aug 8, 2024 · 14e5075 · 14e5075
1 parent d7d227d
commit 14e5075
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src_py/apiServer/stats.py b/src_py/apiServer/stats.py
@@ -104,6 +104,7 @@ def get_loss_ts(self , plot : bool = False , saveToFile : bool = False):
                 plt.savefig('training_loss_function.png', bbox_inches='tight')
 
             plt.show()
+        return df
 
     def get_min_loss(self , plot : bool = False , saveToFile : bool = False): # Todo change it
         """