(self, protocol_version)
| 162 | self.reduce_args = (storage_proxy, *other_reduce_args) |
| 163 | |
| 164 | def __reduce_ex__(self, protocol_version): |
| 165 | if protocol_version != self.protocol_version: |
| 166 | raise RuntimeError( |
| 167 | f'Unexpected protocol version: expected {self.protocol_version}, got {protocol_version}' |
| 168 | ) |
| 169 | return self.reduce_ret_fn, self.reduce_args |
| 170 | |
| 171 | |
| 172 | class IncrementalPyTorchPickler(pickle.Pickler): |