(self, data_type: DataType)
| 663 | return LazyTensor(load, self.shape, data_type, f'convert({data_type}) {self.description}') |
| 664 | |
| 665 | def validate_conversion_to(self, data_type: DataType) -> None: |
| 666 | if data_type != self.data_type and data_type.name not in self.data_type.valid_conversions: |
| 667 | raise ValueError(f'Cannot validate conversion from {self.data_type} to {data_type}.') |
| 668 | |
| 669 | |
| 670 | LazyModel: TypeAlias = 'dict[str, LazyTensor]' |