[skip ci] try not importing rl_cleanup

tlambert03 · Mar 5, 2019 · ff4a80b · ff4a80b
1 parent 3086a1c
commit ff4a80b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/pycudadecon/__init__.py b/pycudadecon/__init__.py
@@ -1,7 +1,7 @@
 from .version import __version__
 from .deconvolution import decon, TemporaryOTF
-from .libcudawrapper import RLContext, rl_decon, rl_init, rl_cleanup
+from .libcudawrapper import RLContext, rl_decon, rl_init
 from .affine import affineGPU, deskewGPU, rotateGPU
 
-__all__ = (decon, TemporaryOTF, RLContext, rl_decon, rl_init, rl_cleanup,
+__all__ = (decon, TemporaryOTF, RLContext, rl_decon, rl_init,
            affineGPU, deskewGPU, rotateGPU)
diff --git a/test/test_decon.py b/test/test_decon.py
@@ -1,7 +1,8 @@
 import unittest
 import os
 import numpy as np
-from pycudadecon import decon, RLContext, rl_init, rl_decon, rl_cleanup
+from pycudadecon import decon, RLContext, rl_init, rl_decon
+from pycudadecon.libcudawrapper import rl_cleanup
 from pycudadecon.util import imread