(self)
| 811 | return lambda index: v[index] |
| 812 | |
| 813 | def get_state_dict(self): |
| 814 | return dict( |
| 815 | config=self.config, |
| 816 | index=self._index, |
| 817 | file_loc=self._file_loc, |
| 818 | total_tokens=self._total_tokens, |
| 819 | ) |
| 820 | |
| 821 | def load_state_dict(self, state_dict): |
| 822 | if 'config' in state_dict: |
nothing calls this directly
no outgoing calls
no test coverage detected