Remove snp_id from data to be added to edge

gyorilab · Dec 12, 2023 · e7af087 · e7af087
1 parent 248b00c
commit e7af087
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/src/indra_cogex/sources/disgenet/__init__.py b/src/indra_cogex/sources/disgenet/__init__.py
@@ -144,8 +144,9 @@ def get_relations(self):  # noqa:D102
         for snp_id, dsi, dpi, disease_name, disease_id, papers in self.df[
             columns
         ].values:
-            data = {"snp_id:str": snp_id, "source": self.name,
-                    "dsi:float": dsi, "dpi:float": dpi,
+            data = {"dsi:float": dsi,
+                    "dpi:float": dpi,
+                    "source": self.name,
                     "papers:int": papers}
             yield Relation(
                 "DBSNP", snp_id, disease_name, disease_id, self.relation,
@@ -181,5 +182,3 @@ def load_disgenet_disease_gene(url, force: bool = False,
         df = df[df["DSI"].notna()]
         df = df[df["DPI"].notna()]
         return df
-
-