From 6e15e9664b0d306d44efa55c56a92b2ada1430e7 Mon Sep 17 00:00:00 2001
From: Simon Alibert <alibert.sim@gmail.com>
Date: Tue, 30 Apr 2024 19:25:11 +0200
Subject: [PATCH] WIP Add save_policy_to_safetensors

---
 tests/scripts/save_policy_to_safetensor.py | 10 ++++++++++
 1 file changed, 10 insertions(+)
 create mode 100644 tests/scripts/save_policy_to_safetensor.py

diff --git a/tests/scripts/save_policy_to_safetensor.py b/tests/scripts/save_policy_to_safetensor.py
new file mode 100644
index 00000000..81748db1
--- /dev/null
+++ b/tests/scripts/save_policy_to_safetensor.py
@@ -0,0 +1,10 @@
+import shutil
+from pathlib import Path
+
+
+def save_policy_to_safetensors(output_dir, repo_id="lerobot/pusht"):
+    ...
+    repo_dir = Path(output_dir) / repo_id
+
+    if repo_dir.exists():
+        shutil.rmtree(repo_dir)