diff --git a/pydra/tasks/nipype1/__init__.py b/pydra/tasks/nipype1/__init__.py index d453018..d786431 100644 --- a/pydra/tasks/nipype1/__init__.py +++ b/pydra/tasks/nipype1/__init__.py @@ -6,3 +6,7 @@ from ._version import __version__ except ImportError: pass + +from .utils import Nipype1Task + +__all__ = ["Nipype1Task"] diff --git a/pydra/tasks/nipype1/tests/test_nipype1task.py b/pydra/tasks/nipype1/tests/test_nipype1task.py index 0c39869..9a17c6e 100644 --- a/pydra/tasks/nipype1/tests/test_nipype1task.py +++ b/pydra/tasks/nipype1/tests/test_nipype1task.py @@ -5,7 +5,7 @@ from nipype.interfaces import fsl import nipype.interfaces.utility as nutil -from pydra.tasks.nipype1.utils import Nipype1Task +from pydra.tasks.nipype1 import Nipype1Task @pytest.mark.skipif(fsl.Info.version() is None, reason="Test requires FSL") diff --git a/pydra/tasks/nipype1/utils.py b/pydra/tasks/nipype1/utils.py index 2946a1d..0be31c4 100644 --- a/pydra/tasks/nipype1/utils.py +++ b/pydra/tasks/nipype1/utils.py @@ -3,6 +3,8 @@ import attrs import typing as ty +__all__ = ["Nipype1Task"] + def traitedspec_to_specinfo(traitedspec): trait_names = set(traitedspec.copyable_trait_names())