mend

mims-harvard · May 13, 2024 · 91ee851 · 91ee851
1 parent 37c14cf
commit 91ee851
Showing 1 changed file with 26 additions and 21 deletions.
diff --git a/tdc/dataset_configs/opentargets_dti.py b/tdc/dataset_configs/opentargets_dti.py
@@ -8,48 +8,53 @@ class OpentargetsDTI(ResourceConfig):
     def __init__(self):
         super(OpentargetsDTI, self).__init__(
             ResourceFeatureGenerator(),
-            keys=["opentargets_ra_data_splits", "opentargets_ibd_data_splits", "df"],
+            keys=[
+                "opentargets_ra_data_splits", "opentargets_ibd_data_splits",
+                "df"
+            ],
             loader_functions=["split", "split", "concat"],
             loader_args=[{
                 "dataset": ("self", "opentargets_ra_drug_evidence"),
                 "column_name":
                     "targetId_genename",
-                "pos_train":
-                    ("self",
-                     ["opentargets_ra_data_splits", "splits", "pos_train_names"]),
+                "pos_train": ("self", [
+                    "opentargets_ra_data_splits", "splits", "pos_train_names"
+                ]),
                 "pos_dev":
                     None,
                 "pos_test":
                     ("self",
-                     ["opentargets_ra_data_splits", "splits", "pos_test_names"]),
-                "neg_train":
-                    ("self",
-                     ["opentargets_ra_data_splits", "splits", "neg_train_names"]),
+                     ["opentargets_ra_data_splits", "splits",
+                      "pos_test_names"]),
+                "neg_train": ("self", [
+                    "opentargets_ra_data_splits", "splits", "neg_train_names"
+                ]),
                 "neg_dev":
                     None,
                 "neg_test":
                     ("self",
-                     ["opentargets_ra_data_splits", "splits", "neg_test_names"]),
+                     ["opentargets_ra_data_splits", "splits",
+                      "neg_test_names"]),
             }, {
                 "dataset": ("self", "opentargets_ibd_drug_evidence"),
                 "column_name":
                     "targetId_genename",
-                "pos_train":
-                    ("self",
-                     ["opentargets_ibd_data_splits", "splits", "pos_train_names"]),
+                "pos_train": ("self", [
+                    "opentargets_ibd_data_splits", "splits", "pos_train_names"
+                ]),
                 "pos_dev":
                     None,
-                "pos_test":
-                    ("self",
-                     ["opentargets_ibd_data_splits", "splits", "pos_test_names"]),
-                "neg_train":
-                    ("self",
-                     ["opentargets_ibd_data_splits", "splits", "neg_train_names"]),
+                "pos_test": ("self", [
+                    "opentargets_ibd_data_splits", "splits", "pos_test_names"
+                ]),
+                "neg_train": ("self", [
+                    "opentargets_ibd_data_splits", "splits", "neg_train_names"
+                ]),
                 "neg_dev":
                     None,
-                "neg_test":
-                    ("self",
-                     ["opentargets_ibd_data_splits", "splits", "neg_test_names"])
+                "neg_test": ("self", [
+                    "opentargets_ibd_data_splits", "splits", "neg_test_names"
+                ])
             }, {
                 "ds_list": [
                     "opentargets_ibd_data_splits", "opentargets_ra_data_splits"