Merge pull request #27 from ghisvail/enh/public-api

[ENH] Expose Nipype1Task to the root package
nipype · Nov 21, 2022 · cea8f9b · cea8f9b
2 parents 536458e + 69f0444
commit cea8f9b
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 1 deletion.
diff --git 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
@@ -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
@@ -3,6 +3,8 @@
 import attrs
 import typing as ty
 
+__all__ = ["Nipype1Task"]
+
 
 def traitedspec_to_specinfo(traitedspec):
     trait_names = set(traitedspec.copyable_trait_names())