From 85d3db61e137c68192a6fa518ed10eb822827dbb Mon Sep 17 00:00:00 2001
From: Max Jones <meghanj@alum.mit.edu>
Date: Wed, 4 May 2022 14:42:55 -0400
Subject: [PATCH] Mock up torch imports

---
 doc/conf.py         | 1 +
 doc/environment.yml | 1 -
 2 files changed, 1 insertion(+), 1 deletion(-)

diff --git a/doc/conf.py b/doc/conf.py
index ef5de36..bd016e5 100644
--- a/doc/conf.py
+++ b/doc/conf.py
@@ -70,6 +70,7 @@ def skip(app, what, name, obj, skip, options):
 def setup(app):
     app.connect('autodoc-skip-member', skip)
 
+autodoc_mock_imports = ["torch"]
 
 # link to github issues
 extlinks = {
diff --git a/doc/environment.yml b/doc/environment.yml
index c84a740..199c31e 100644
--- a/doc/environment.yml
+++ b/doc/environment.yml
@@ -12,7 +12,6 @@ dependencies:
   - numpy
   - numpydoc
   - pytest
-  - pytorch
   - sphinx
   - sphinx-autosummary-accessors
   - sphinx-copybutton