make gpax and sable optional dependencies

README.md

@@ -16,9 +16,12 @@ To install Dial from source, git clone this repository and run the following fro
 
 `pip install .`
 
-Alternatively, both intersect-sdk and Dial may be installed with the following:
+By default, **only scikit-learn is available as a backend**. We also support GPax and Sable as backends. To install these:
 
-`pip install -e .`
+- for GPax: `pip install ".[gpax]"`
+- for Sable: `pip install ".[sable]"`
+
+Use commas to add multiple backends: `pip install ".[gpax,sable]"`
 
 ## Installing (developers)
 

pyproject.toml

@@ -20,12 +20,9 @@ classifiers = ["Programming Language :: Python :: 3"]
 dependencies = [
   "intersect_sdk>=0.9.3,<0.10.0",
   "numpy",
-  "scikit-learn>=1.4.0,<2.0.0",                        # TODO consider making an optional dependency group
+  "scikit-learn>=1.4.0,<2.0.0",   # TODO consider making an optional dependency group
   "scipy>=1.12.0,<2.0.0",
-  "gpax>=0.1.8",                                       # TODO consider making an optional dependency group
-  "numpyro<0.20.1",                                    # depend on older numpyro for gpax (numpyro.contrib.module: random_haiku_module)
-  "pymongo>=4.12.1",                                   # TODO - this is only needed for dial_service, dial_dataclass can use a simple fixture for ObjectID representation which allows it to skip this dependency
-  "sable @ git+https://code.ornl.gov/sable/sable.git", # TODO this references the internal repo, transition to public version
+  "pymongo>=4.12.1",              # TODO - this is only needed for dial_service, dial_dataclass can use a simple fixture for ObjectID representation which allows it to skip this dependency
 ]
 
 [project.urls]
@@ -36,6 +33,11 @@ Issues = "https://github.com/INTERSECT-DIAL/dial/issues"
 
 [project.optional-dependencies]
 docs = ["sphinx>=5.3.0", "furo>=2023.3.27"]
+gpax = [
+  "gpax>=0.1.8",
+  "numpyro<0.20.1", # depend on older numpyro for gpax (numpyro.contrib.module: random_haiku_module)
+]
+sable = ["sable @ git+https://code.ornl.gov/sable/sable.git"]
 
 [dependency-groups]
 dev = [