docs: fix cufinufft autodoc

flatironinstitute · Aug 22, 2023 · 56ace33 · 56ace33
1 parent e283579
commit 56ace33
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/docs/conf.py b/docs/conf.py
@@ -25,7 +25,7 @@
 sys.path.insert(0, os.path.abspath('../python/cufinufft'))
 
 # quash sphinx's inevitable failure of C++ lib extension to import...
-autodoc_mock_imports = ['finufft._finufft', 'numpy']
+autodoc_mock_imports = ['finufft._finufft', 'numpy', 'pycuda']
 # The above is not enough for nested import -- forcibly mock them out ahead of time:
 #for name in autodoc_mock_imports:
 #    sys.modules[name] = sphinx.ext.autodoc._MockModule(name, None)

diff --git a/python/cufinufft/cufinufft/__init__.py b/python/cufinufft/cufinufft/__init__.py
@@ -3,5 +3,9 @@
 from cufinufft._simple import (nufft1d1, nufft1d2, nufft2d1, nufft2d2,
                                nufft3d1, nufft3d2)
 
-__all__ = ['cufinufft']
+__all__ = ["nufft1d1", "nufft1d2",
+           "nufft2d1", "nufft2d2",
+           "nufft3d1", "nufft3d2",
+           "Plan"]
+
 __version__ = '2.2.0dev0'