@@ -495,7 +495,7 @@ async def _get_mapped_nodeports_values(
495495 PORTS : Nodeports = await node_ports_v2 .ports (
496496 user_id = user_id ,
497497 project_id = ProjectIDStr (project_id ),
498- node_uuid = NodeIDStr (node_uuid ),
498+ node_uuid = TypeAdapter ( NodeIDStr ). validate_python (node_uuid ),
499499 db_manager = db_manager ,
500500 )
501501 result [str (node_uuid )] = InputsOutputs (
@@ -699,7 +699,7 @@ async def _fetch_data_via_aioboto(
699699 r_clone_settings : RCloneSettings ,
700700 dir_tag : str ,
701701 temp_dir : Path ,
702- node_id : NodeID ,
702+ node_id : NodeIDStr ,
703703 project_id : ProjectID ,
704704) -> Path :
705705 save_to = temp_dir / f"aioboto_{ dir_tag } _{ uuid4 ()} "
@@ -840,7 +840,7 @@ async def _assert_push_non_file_outputs(
840840 logger .debug ("Going to poll task %s" , task_id )
841841
842842 async def _debug_progress_callback (
843- message : ProgressMessage , percent : ProgressPercent , task_id : TaskId
843+ message : ProgressMessage , percent : ProgressPercent | None , task_id : TaskId
844844 ) -> None :
845845 logger .debug ("%s: %.2f %s" , task_id , percent , message )
846846
@@ -1118,7 +1118,7 @@ async def test_nodeports_integration(
11181118 dir_tag = "dy" ,
11191119 user_id = current_user ["id" ],
11201120 project_id = current_study .uuid ,
1121- service_uuid = services_node_uuids .dy ,
1121+ service_uuid = NodeID ( services_node_uuids .dy ) ,
11221122 temp_dir = tmp_path ,
11231123 io_log_redirect_cb = mock_io_log_redirect_cb ,
11241124 faker = faker ,
@@ -1139,7 +1139,7 @@ async def test_nodeports_integration(
11391139 dir_tag = "dy_compose_spec" ,
11401140 user_id = current_user ["id" ],
11411141 project_id = current_study .uuid ,
1142- service_uuid = services_node_uuids .dy_compose_spec ,
1142+ service_uuid = NodeID ( services_node_uuids .dy_compose_spec ) ,
11431143 temp_dir = tmp_path ,
11441144 io_log_redirect_cb = mock_io_log_redirect_cb ,
11451145 faker = faker ,
0 commit comments