Create test_dataset

test_dataset

@@ -0,0 +1,17 @@
+from chroma_datasets.utils import import_into_chroma
+from chroma_datasets.utils import load_huggingface_dataset, import_chroma_exported_hf_dataset
+import huggingface_hub
+
+
+UB_paragraphs_dataset_name = "CenterFor/UB_paragraphs"
+UB_paragraphs_dataset = load_huggingface_dataset(UB_paragraphs_dataset_name)
+print(UB_paragraphs_dataset)
+
+chroma_collection_name_from_HF = "entire_book_paragraph_content-from-HF"
+paragraph_content_collection_from_HF = import_chroma_exported_hf_dataset(
+    chroma_client=chroma_client, 
+    dataset=UB_paragraphs_dataset, 
+    collection_name=chroma_collection_name_from_HF,
+    embedding_function=chroma_embedding_function
+)
+print(paragraph_content_collection_from_HF.count())