2626
2727
2828if ty .TYPE_CHECKING :
29- from pydra .engine .specs import TaskSpec , Outputs
29+ from pydra .engine .specs import TaskSpec , TaskOutputs
3030 from pydra .engine .core import Task
3131
3232__all__ = [
@@ -351,7 +351,7 @@ def get_fields(klass, field_type, auto_attribs, helps) -> dict[str, Field]:
351351
352352def make_task_spec (
353353 spec_type : type ["TaskSpec" ],
354- out_type : type ["Outputs " ],
354+ out_type : type ["TaskOutputs " ],
355355 task_type : type ["Task" ],
356356 inputs : dict [str , Arg ],
357357 outputs : dict [str , Out ],
@@ -467,11 +467,11 @@ def make_task_spec(
467467
468468
469469def make_outputs_spec (
470- spec_type : type ["Outputs " ],
470+ spec_type : type ["TaskOutputs " ],
471471 outputs : dict [str , Out ],
472472 bases : ty .Sequence [type ],
473473 spec_name : str ,
474- ) -> type ["Outputs " ]:
474+ ) -> type ["TaskOutputs " ]:
475475 """Create an outputs specification class and its outputs specification class from the
476476 output fields provided to the decorator/function.
477477
@@ -492,10 +492,10 @@ def make_outputs_spec(
492492 klass : type
493493 The class created using the attrs package
494494 """
495- from pydra .engine .specs import Outputs
495+ from pydra .engine .specs import TaskOutputs
496496
497497 if not any (issubclass (b , spec_type ) for b in bases ):
498- if out_spec_bases := [b for b in bases if issubclass (b , Outputs )]:
498+ if out_spec_bases := [b for b in bases if issubclass (b , TaskOutputs )]:
499499 raise ValueError (
500500 f"Cannot make { spec_type } output spec from { out_spec_bases } bases"
501501 )
0 commit comments