added finetune_embeddings_tests to main for testing

StampyAI · Aug 16, 2023 · 877bcfe · 877bcfe
1 parent 7bd986a
commit 877bcfe
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/main.py b/main.py
@@ -124,6 +124,14 @@ def pinecone_update_all(self, *skip) -> None:
         names = [name for name in ALL_DATASETS if name not in skip]
         PineconeUpdater().update(names)
 
+    def finetune_embeddings_tests(self) -> None:
+        """
+        This function tests the finetune_embeddings function
+        """
+        from align_data.finetuning.finetune_model import finetune_embeddings
+
+        finetune_embeddings()
+
 
 if __name__ == "__main__":
     fire.Fire(AlignmentDataset)