tests

canals/pipeline/pipeline.py

@@ -22,7 +22,7 @@
     PipelineValidationError,
 )
 from canals.pipeline.draw import _draw, _convert_for_debug, RenderingEngines
-from canals.pipeline.validation import _validate_pipeline_input, _find_pipeline_inputs
+from canals.pipeline.validation import validate_pipeline_input, _find_pipeline_inputs
 from canals.pipeline.connections import parse_connection, _find_unambiguous_connection
 from canals.type_utils import _type_name
 from canals.serialization import component_to_dict, component_from_dict

canals/pipeline/validation.py

@@ -7,13 +7,14 @@
 import networkx
 
 from canals.errors import PipelineValidationError
+from canals.component.sockets import InputSocket
 from canals.pipeline.descriptions import _find_pipeline_inputs, _describe_pipeline_inputs_as_string
 
 
 logger = logging.getLogger(__name__)
 
 
-def _validate_pipeline_input(graph: networkx.MultiDiGraph, input_values: Dict[str, Any]) -> Dict[str, Any]:
+def validate_pipeline_input(graph: networkx.MultiDiGraph, input_values: Dict[str, Any]) -> Dict[str, Any]:
     """
     Make sure the pipeline is properly built and that the input received makes sense.
     Returns the input values, validated and updated at need.