fml

cpnota · Jan 14, 2020 · cae9f0d · cae9f0d
1 parent 88c0049
commit cae9f0d
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 2 deletions.
diff --git a/.readthedocs.yml b/.readthedocs.yml
@@ -24,4 +24,3 @@ python:
       path: .
       extra_requirements:
         - docs
-        - pytorch
diff --git a/all/approximation/q_dist.py b/all/approximation/q_dist.py
@@ -56,7 +56,7 @@ def project(self, dist, support):
         return target_dist
 
 
-class QDistModule(nn.Module):
+class QDistModule(torch.nn.Module):
     def __init__(self, model, n_actions, atoms):
         super().__init__()
         self.atoms = atoms

diff --git a/docs/source/conf.py b/docs/source/conf.py
@@ -42,6 +42,12 @@
 autodoc_default_flags = ['members']
 autosummary_generate = True
 
+# Mock requirements to save resources during doc build machine setup
+autodoc_mock_imports = [
+    'torch',
+    'torchvision',
+]
+
 # Add any paths that contain templates here, relative to this directory.
 templates_path = ['_templates']