From 56ace33188952a6add3e55ff6ab6750135c5c095 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Joakim=20And=C3=A9n?= Date: Mon, 31 Jul 2023 23:51:08 -0500 Subject: [PATCH] docs: fix cufinufft autodoc --- docs/conf.py | 2 +- python/cufinufft/cufinufft/__init__.py | 6 +++++- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/docs/conf.py b/docs/conf.py index 1fff8a678..602b7c80d 100644 --- 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 index 730f6a693..de79ff315 100644 --- 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'