Merge pull request #10 from rishiraj/staging

Staging
rishiraj · Apr 14, 2024 · 51ecc20 · 51ecc20
2 parents 4e91fa0 + e10a373
commit 51ecc20
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/spanking/main.py b/spanking/main.py
@@ -20,7 +20,7 @@ def add_texts(self, texts):
     def delete_text(self, index):
         if 0 <= index < len(self.texts):
             self.texts.pop(index)
-            self.embeddings = self.embeddings.at[index].delete()
+            self.embeddings = jnp.delete(self.embeddings, index, axis=0)
         else:
             raise IndexError("Invalid index")