Merge pull request #24 from cthoyt/resolve-datasets

Add dataset resolver around dataset loader class

Author: benedekrozemberczki

chemicalx/data/__init__.py

@@ -1,6 +1,10 @@
+from class_resolver import Resolver
+
 from .contextfeatureset import *  # noqa:F401,F403
 from .drugfeatureset import *  # noqa:F401,F403
 from .labeledtriples import *  # noqa:F401,F403
 from .datasetloader import *  # noqa:F401,F403
 from .drugpairbatch import *  # noqa:F401,F403
 from .batchgenerator import *  # noqa:F401,F403
+
+dataset_resolver = Resolver.from_subclasses(base=DatasetLoader)

chemicalx/data/datasetloader.py

@@ -6,6 +6,12 @@
 from typing import Dict
 from chemicalx.data import DrugFeatureSet, ContextFeatureSet, LabeledTriples
 
+__all__ = [
+    "DatasetLoader",
+    "DrugCombDbDatasetLoader",
+    "DrugCombDatasetLoader",
+]
+
 
 class DatasetLoader:
     """
@@ -102,3 +108,18 @@ def get_labeled_triples(self):
         labeled_triples = LabeledTriples()
         labeled_triples.update_from_pandas(raw_data)
         return labeled_triples
+
+
+class DrugCombDbDatasetLoader(DatasetLoader):
+    """A dataset loader for `DrugCombDB <http://drugcombdb.denglab.org>`_."""
+
+    def __init__(self):
+        super().__init__("drugcombdb")
+
+
+# FIXME need reference
+class DrugCombDatasetLoader(DatasetLoader):
+    """A dataset loader for DrugComb."""
+
+    def __init__(self):
+        super().__init__("drugcomb")