ddi-fw 0.0.47__py3-none-any.whl → 0.0.48__py3-none-any.whl

ddi_fw/datasets/core.py

@@ -379,13 +379,13 @@ class BaseDataset(ABC):
                 # return np.hstack(
                 #     (value[row['id1']], value[row['id2']]), dtype=np.float16)
 
-        def x_fnc(row, embedding_column, embeddings_after_pooling):
-            if row['id1'] in self.embedding_dict[embedding_column]:
-                v1 = embeddings_after_pooling[embedding_column][row['id1']]
+        def x_fnc(row, embeddings_after_pooling):
+            if row['id1'] in embeddings_after_pooling:
+                v1 = embeddings_after_pooling[row['id1']]
             else:
                 v1 = np.zeros(self.embedding_size)
-            if row['id2'] in self.embedding_dict[embedding_column]:
-                v2 = embeddings_after_pooling[embedding_column][row['id2']]
+            if row['id2'] in embeddings_after_pooling:
+                v2 = embeddings_after_pooling[row['id2']]
             else:
                 v2 = np.zeros(self.embedding_size)
             return np.float16(np.hstack(
@@ -404,7 +404,7 @@ class BaseDataset(ABC):
                 v) for k, v in self.embedding_dict[embedding_column].items()}
             # column_embeddings_dict = embedding_values[embedding_column]
             self.ddis_df[embedding_column+'_embedding'] = self.ddis_df.apply(
-                x_fnc, args=(embedding_column, embeddings_after_pooling), axis=1)
+                x_fnc, args=(embeddings_after_pooling,), axis=1)
 
         self.dataframe = self.ddis_df.copy()
         self.dataframe['class'] = list(classes)

{ddi_fw-0.0.47.dist-info → ddi_fw-0.0.48.dist-info}/METADATA

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

{ddi_fw-0.0.47.dist-info → ddi_fw-0.0.48.dist-info}/RECORD

@@ -1,5 +1,5 @@
 ddi_fw/datasets/__init__.py,sha256=CqDrx7Ov83pXRh-n0ylembBmzhlW_yFWiheBcISrKdg,510
-ddi_fw/datasets/core.py,sha256=fnpoCJbVEBPhy8Hvp0WJUfQw9nROPL29oRy2yGz30SI,19029
+ddi_fw/datasets/core.py,sha256=CFmnK0_cUxQAlTHjC2LEuKX0DNVAhGjSyQaV4jURYdI,18932
 ddi_fw/datasets/db_utils.py,sha256=OTsa3d-Iic7z3HmzSQK9UigedRbHDxYChJk0s4GfLnw,6191
 ddi_fw/datasets/embedding_generator.py,sha256=Jqrlv88RCu0Lg812KsA12X0cSaZuxbckJ4LNRKNy_qw,2173
 ddi_fw/datasets/embedding_generator_new.py,sha256=bdZ7dSrm0oJTRh84MyltB_hhgL_IQyvusO6aPfW8JxU,7257
@@ -80,7 +80,7 @@ ddi_fw/utils/enums.py,sha256=19eJ3fX5eRK_xPvkYcukmug144jXPH4X9zQqtsFBj5A,671
 ddi_fw/utils/py7zr_helper.py,sha256=rWhz3Trj6bk6B24hMuff1fKdRbC0yuvaHFlL8-eaAhg,4450
 ddi_fw/utils/utils.py,sha256=Na6Y8mY-CFbQjrgd9xC8agcrjVvTj_7KIXqFm1H_3qU,3549
 ddi_fw/utils/zip_helper.py,sha256=YRZA4tKZVBJwGQM0_WK6L-y5MoqkKoC-nXuuHK6CU9I,5567
-ddi_fw-0.0.47.dist-info/METADATA,sha256=0eMpCL29UFk82ma89yww9jABXzP5QE3tPaYP6oIOxCg,1565
-ddi_fw-0.0.47.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
-ddi_fw-0.0.47.dist-info/top_level.txt,sha256=PMwHICFZTZtcpzQNPV4UQnfNXYIeLR_Ste-Wfc1h810,7
-ddi_fw-0.0.47.dist-info/RECORD,,
+ddi_fw-0.0.48.dist-info/METADATA,sha256=NT2xu52EFUO2Rtk4hDZtCftemzeHYBTLZ7ktOHnEUxM,1565
+ddi_fw-0.0.48.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+ddi_fw-0.0.48.dist-info/top_level.txt,sha256=PMwHICFZTZtcpzQNPV4UQnfNXYIeLR_Ste-Wfc1h810,7
+ddi_fw-0.0.48.dist-info/RECORD,,