Clear data after plot is generated

lavague-ai · Aug 26, 2024 · bcbfcdb · bcbfcdb
1 parent 0bf4fbf
commit bcbfcdb
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/lavague-core/lavague/core/agents.py b/lavague-core/lavague/core/agents.py
@@ -609,4 +609,6 @@ def get_summary(self):
         plot = chart_generator.plot_waterfall()
         table = chart_generator.get_summary_df()
 
+        clear_profiling_data()
+
         return plot, table