(self, kwargs)
| 439 | self._pipe._cur_iter_batch_info.mark_non_uniform_batch() |
| 440 | |
| 441 | def _separate_kwargs(self, kwargs): |
| 442 | self._init_args = {} |
| 443 | self._call_args = {} |
| 444 | self._kwargs_classification = {} |
| 445 | |
| 446 | for key, value in kwargs.items(): |
| 447 | classification = _Classification( |
| 448 | value, f"Argument {key}", arg_constant_len=self._batch_size |
| 449 | ) |
| 450 | if classification.is_batch: |
| 451 | self._call_args[key] = classification.data |
| 452 | else: |
| 453 | self._init_args[key] = classification.data |
| 454 | self._kwargs_classification[key] = classification |
| 455 | |
| 456 | def _init_arithm_op(self, name, inputs): |
| 457 | """Fills arithmetic operator init arguments and returns inputs that are DataNodes.""" |
no test coverage detected