@@ -28,7 +28,8 @@ async def pull_user_services_docker_images_task(
2828 ),
2929 lrt_namespace = lrt_namespace ,
3030 )
31- return TaskId (result )
31+ assert isinstance (result , TaskId ) # nosec
32+ return result
3233
3334
3435@log_decorator (_logger , level = logging .DEBUG )
@@ -46,7 +47,8 @@ async def create_service_containers_task(
4647 lrt_namespace = lrt_namespace ,
4748 containers_create = containers_create ,
4849 )
49- return TaskId (result )
50+ assert isinstance (result , TaskId ) # nosec
51+ return result
5052
5153
5254@log_decorator (_logger , level = logging .DEBUG )
@@ -62,7 +64,8 @@ async def runs_docker_compose_down_task(
6264 TypeAdapter (RPCMethodName ).validate_python ("runs_docker_compose_down_task" ),
6365 lrt_namespace = lrt_namespace ,
6466 )
65- return TaskId (result )
67+ assert isinstance (result , TaskId ) # nosec
68+ return result
6669
6770
6871@log_decorator (_logger , level = logging .DEBUG )
@@ -78,7 +81,8 @@ async def state_restore_task(
7881 TypeAdapter (RPCMethodName ).validate_python ("state_restore_task" ),
7982 lrt_namespace = lrt_namespace ,
8083 )
81- return TaskId (result )
84+ assert isinstance (result , TaskId ) # nosec
85+ return result
8286
8387
8488@log_decorator (_logger , level = logging .DEBUG )
@@ -94,7 +98,8 @@ async def state_save_task(
9498 TypeAdapter (RPCMethodName ).validate_python ("state_save_task" ),
9599 lrt_namespace = lrt_namespace ,
96100 )
97- return TaskId (result )
101+ assert isinstance (result , TaskId ) # nosec
102+ return result
98103
99104
100105@log_decorator (_logger , level = logging .DEBUG )
@@ -112,7 +117,8 @@ async def ports_inputs_pull_task(
112117 lrt_namespace = lrt_namespace ,
113118 port_keys = port_keys ,
114119 )
115- return TaskId (result )
120+ assert isinstance (result , TaskId ) # nosec
121+ return result
116122
117123
118124@log_decorator (_logger , level = logging .DEBUG )
@@ -130,7 +136,8 @@ async def ports_outputs_pull_task(
130136 lrt_namespace = lrt_namespace ,
131137 port_keys = port_keys ,
132138 )
133- return TaskId (result )
139+ assert isinstance (result , TaskId ) # nosec
140+ return result
134141
135142
136143@log_decorator (_logger , level = logging .DEBUG )
@@ -146,7 +153,8 @@ async def ports_outputs_push_task(
146153 TypeAdapter (RPCMethodName ).validate_python ("ports_outputs_push_task" ),
147154 lrt_namespace = lrt_namespace ,
148155 )
149- return TaskId (result )
156+ assert isinstance (result , TaskId ) # nosec
157+ return result
150158
151159
152160@log_decorator (_logger , level = logging .DEBUG )
@@ -162,4 +170,5 @@ async def containers_restart_task(
162170 TypeAdapter (RPCMethodName ).validate_python ("containers_restart_task" ),
163171 lrt_namespace = lrt_namespace ,
164172 )
165- return TaskId (result )
173+ assert isinstance (result , TaskId ) # nosec
174+ return result
0 commit comments