expand mock

BasisResearch · Nov 10, 2021 · 823fead · 823fead
1 parent dcf5498
commit 823fead
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/docs/source/conf.py b/docs/source/conf.py
@@ -75,7 +75,7 @@
 # }
 
 MOCK_MODULES = ['numpy', 'pandas', 'torch', 'polyagamma', 'torch.distributions',
-                'torch.linalg', 'torch.nn.functional']
+                'torch.linalg', 'torch.nn.functional', 'functools']
 
 for mod_name in MOCK_MODULES:
    sys.modules[mod_name] = mock.Mock()

diff --git a/setup.py b/setup.py
@@ -49,7 +49,6 @@
         "License :: OSI Approved :: Apache Software License",
         "Operating System :: POSIX :: Linux",
         "Operating System :: MacOS :: MacOS X",
-        "Programming Language :: Python :: 3.6",
-        "Programming Language :: Python :: 3.7",
+        "Programming Language :: Python :: 3.8",
     ],
 )