pharmone map

tdc/metadata.py

@@ -940,6 +940,22 @@ def get_task2category():
     "geneformer_gene_median_dictionary": "pkl",
     "geneformer_gene_name_id_dict": "pkl",
     "geneformer_token_dictionary": "pkl",
+    "evebio_pharmone_v1_assay_doc": "tab",
+    "evebio_pharmone_v1_assay_table": "tab",
+    "evebio_pharmone_v1_bundle_doc": "tab",
+    "evebio_pharmone_v1_bundle_table": "tab",
+    "evebio_pharmone_v1_compound_doc": "tab",
+    "evebio_pharmone_v1_compound_table": "tab",
+    "evebio_pharmone_v1_control_doc": "tab",
+    "evebio_pharmone_v1_control_table": "tab",
+    "evebio_pharmone_v1_detailed_result_doc": "tab",
+    "evebio_pharmone_v1_detailed_result_table": "tab",
+    "evebio_pharmone_v1_observed_points_doc": "tab",
+    "evebio_pharmone_v1_observed_points_table": "tab",
+    "evebio_pharmone_v1_summary_result_doc": "tab",
+    "evebio_pharmone_v1_summary_result_table": "tab",
+    "evebio_pharmone_v1_target_doc": "tab",
+    "evebio_pharmone_v1_target_table": "tab",
 }
 
 name2id = {
@@ -1130,6 +1146,22 @@ def get_task2category():
     "geneformer_gene_median_dictionary": 10626278,
     "geneformer_gene_name_id_dict": 10626276,
     "geneformer_token_dictionary": 10626277,
+    "evebio_pharmone_v1_assay_doc": 10741530,
+    "evebio_pharmone_v1_assay_table": 10741541,
+    "evebio_pharmone_v1_bundle_doc": 10741540,
+    "evebio_pharmone_v1_bundle_table": 10741543,
+    "evebio_pharmone_v1_compound_doc": 10741535,
+    "evebio_pharmone_v1_compound_table": 10741532,
+    "evebio_pharmone_v1_control_doc": 10741539,
+    "evebio_pharmone_v1_control_table": 10741538,
+    "evebio_pharmone_v1_detailed_result_doc": 10741545,
+    "evebio_pharmone_v1_detailed_result_table": 10741534,
+    "evebio_pharmone_v1_observed_points_doc": 10741531,
+    "evebio_pharmone_v1_observed_points_table": 10741544,
+    "evebio_pharmone_v1_summary_result_doc": 10741529,
+    "evebio_pharmone_v1_summary_result_table": 10741542,
+    "evebio_pharmone_v1_target_doc": 10741536,
+    "evebio_pharmone_v1_target_table": 10741537,
 }
 
 oracle2type = {

tdc/resource/pharmone.py

@@ -0,0 +1,45 @@
+from ..utils.load import download_wrapper
+
+"""
+Resource class for the Eve Bio (https://evebio.org/) Pharmone Map.
+"""
+
+class PharmoneMap(object):
+
+    def __init__(self, path="./data"):
+        self.path = path
+
+    def get_data(self):
+        return download_wrapper(
+            'evebio_pharmone_v1_detailed_result_table',
+            self.path,
+            'evebio_pharmone_v1_detailed_result_table'
+        )  # Load the Pharmone Map data
+
+    def get_obs_metadata(self):
+        return download_wrapper(
+            "evebio_pharmone_v1_observed_points_table",
+            self.path,
+            "evebio_pharmone_v1_observed_points_table"
+        )
+
+    def get_control_data(self):
+        return download_wrapper(
+            "evebio_pharmone_v1_control_table",
+            self.path,
+            "evebio_pharmone_v1_control_table"
+        )  # Load the control data
+
+    def get_compound_data(self):
+        return download_wrapper(
+            "evebio_pharmone_v1_compound_table",
+            self.path,
+            "evebio_pharmone_v1_compound_table"
+        )
+
+    def get_target_data(self):
+        return download_wrapper(
+            "evebio_pharmone_v1_target_table",
+            self.path,
+            "evebio_pharmone_v1_target_table"
+        )

tdc/test/test_resources.py

@@ -125,6 +125,30 @@ def tearDown(self):
         except:
             pass
 
+class TestPharmoneMap(unittest.TestCase):
+
+    def setUp(self):
+        print(os.getcwd())
+        pass
+
+    def test_get_data(self):
+        from tdc.resource.pharmone import PharmoneMap
+
+        resource = PharmoneMap()
+        data = resource.get_data()
+        assert data
+        assert isinstance(data, DataFrame)
+        assert "Compound" in data.columns
+        assert "Target_ID" in data.columns
+        assert "pXC50" in data.columns
+
+    def tearDown(self):
+        try:
+            print(os.getcwd())
+            shutil.rmtree(os.path.join(os.getcwd(), "data"))
+        except:
+            pass
+
 
 if __name__ == "__main__":
     unittest.main()