From 839f3c4142f2c143064fcdec5262aa0c8b418863 Mon Sep 17 00:00:00 2001
From: Brandon Hancock <brandon@brandonhancock.io>
Date: Tue, 28 Jan 2025 14:27:57 -0500
Subject: [PATCH] add clear

---
 src/crewai/utilities/training_handler.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/src/crewai/utilities/training_handler.py b/src/crewai/utilities/training_handler.py
index 5cadde619..b6b3c38b6 100644
--- a/src/crewai/utilities/training_handler.py
+++ b/src/crewai/utilities/training_handler.py
@@ -1,3 +1,5 @@
+import os
+
 from crewai.utilities.file_handler import PickleHandler
 
 
@@ -29,3 +31,10 @@ class CrewTrainingHandler(PickleHandler):
             data[agent_id] = {train_iteration: new_data}
 
         self.save(data)
+
+    def clear(self) -> None:
+        """Clear the training data by removing the file or resetting its contents."""
+        if os.path.exists(self.file_path):
+            with open(self.file_path, "wb") as file:
+                # Overwrite with an empty dictionary
+                self.save({})