diff --git a/Python/pyproject.toml b/Python/pyproject.toml
index bef4b1a..d889cf7 100644
--- a/Python/pyproject.toml
+++ b/Python/pyproject.toml
@@ -19,6 +19,9 @@ dependencies = [
     'tqdm'
 ]
 
+[project.optional-dependencies]
+docs = ["mock"]
+
 [[project.authors]]
 name = "Joshua Pickard"
 email = "jpic@umich.edu"
diff --git a/docs/conf.py b/docs/conf.py
index c3665b6..ca3c2e3 100644
--- a/docs/conf.py
+++ b/docs/conf.py
@@ -16,7 +16,8 @@
                 'pandas',
                 'pd',
                 'np',
-                'sp'
+                'sp',
+                'mock'
                ]
 for mod_name in MOCK_MODULES:
     sys.modules[mod_name] = mock.Mock()