| 406 | @arg_handler(handler=select_args_handler) |
| 407 | @trace_user_frame |
| 408 | def select(self, *args: pw.ColumnReference, **kwargs: Any) -> pw.Table: |
| 409 | if self._mode not in [pw.JoinMode.LEFT, pw.JoinMode.RIGHT, pw.JoinMode.OUTER]: |
| 410 | raise ValueError(f"Unsupported asof join mode: {self._mode}") |
| 411 | |
| 412 | expressions: dict[str, expr.ColumnExpression] = {} |
| 413 | |
| 414 | all_args = combine_args_kwargs(args, kwargs) |
| 415 | |
| 416 | for new_name, expression in all_args.items(): |
| 417 | expressions[new_name] = self._sub_desugaring.eval_expression(expression) |
| 418 | return self._merge_result.select(**expressions) |
| 419 | |
| 420 | |
| 421 | def _asof_join( |