ddi-fw 0.0.196__tar.gz → 0.0.197__tar.gz

{ddi_fw-0.0.196 → ddi_fw-0.0.197}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ddi_fw
-Version: 0.0.196
+Version: 0.0.197
 Summary: Do not use :)
 Author-email: Kıvanç Bayraktar <bayraktarkivanc@gmail.com>
 Maintainer-email: Kıvanç Bayraktar <bayraktarkivanc@gmail.com>

{ddi_fw-0.0.196 → ddi_fw-0.0.197}/pyproject.toml

@@ -6,7 +6,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "ddi_fw"
-version = "0.0.196"
+version = "0.0.197"
 description = "Do not use :)"
 readme = "README.md"
 authors = [

{ddi_fw-0.0.196 → ddi_fw-0.0.197}/src/ddi_fw/datasets/core.py

@@ -135,7 +135,7 @@ class BaseDataset(BaseModel):
         skip deriving them. Otherwise, derive them from the dataframe and indices.
         """
         self.prep()
-        
+
         if isinstance(self, TextDatasetMixin):
             self.process_text()
 
@@ -267,6 +267,9 @@ class TextDatasetMixin(BaseModel):
     vector_db_persist_directory: Optional[str] = None
     vector_db_collection_name: Optional[str] = None
 
+    class Config:
+        arbitrary_types_allowed = True
+
     def __create_or_update_embeddings__(self, embedding_dict, vector_db_persist_directory, vector_db_collection_name, column=None):
         """
         Fetch embeddings and metadata from a persistent Chroma vector database and update the provided embedding_dict.
@@ -314,14 +317,13 @@ class TextDatasetMixin(BaseModel):
         else:
             raise ValueError(
                 "Persistent directory for the vector DB is not specified.")
-    
+
     def process_text(self):
         # key, value = next(iter(embedding_dict.items()))
         #     embedding_size = value[next(iter(value))][0].shape[0]
         #     pooling_strategy = self.embedding_pooling_strategy_type(
-        #     ) if self.embedding_pooling_strategy_type else None   
-           
-           
+        #     ) if self.embedding_pooling_strategy_type else None
+
         # 'enzyme','target','pathway','smile','all_text','indication', 'description','mechanism_of_action','pharmacodynamics', 'tui', 'cui', 'entities'
         # kwargs = {"columns": self.columns}
         # if self.ner_threshold:

{ddi_fw-0.0.196 → ddi_fw-0.0.197}/src/ddi_fw.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ddi_fw
-Version: 0.0.196
+Version: 0.0.197
 Summary: Do not use :)
 Author-email: Kıvanç Bayraktar <bayraktarkivanc@gmail.com>
 Maintainer-email: Kıvanç Bayraktar <bayraktarkivanc@gmail.com>