(args)
| 271 | |
| 272 | |
| 273 | def extract_grouped_input_state_callback_args_from_args(args): |
| 274 | # Collect input and state from args |
| 275 | parameters = [] |
| 276 | while args: |
| 277 | next_deps = flatten_grouping(args[0]) |
| 278 | if all(isinstance(d, (Input, State)) for d in next_deps): |
| 279 | parameters.append(args.pop(0)) |
| 280 | else: |
| 281 | break |
| 282 | |
| 283 | if len(parameters) == 1: |
| 284 | # Only one output grouping, return as-is |
| 285 | return parameters[0] |
| 286 | |
| 287 | # Multiple output groupings, return wrap in tuple |
| 288 | return parameters |
| 289 | |
| 290 | |
| 291 | def extract_grouped_input_state_callback_args(args, kwargs): |
no test coverage detected
searching dependent graphs…