(file_name)
| 752 | |
| 753 | |
| 754 | def _load_selected_rows(file_name): |
| 755 | temp_sr = core.SelectedRows() |
| 756 | if _is_file_path(file_name): |
| 757 | # '_seek' is the end position of this SelectedRows in the file. |
| 758 | _seek = core.load_selected_rows(temp_sr, file_name) |
| 759 | |
| 760 | elif _is_memory_buffer(file_name): |
| 761 | with _open_file_buffer(file_name, 'rb') as f: |
| 762 | selected_rows_bytes = f.read() |
| 763 | paddle.base.core.load_selected_rows_from_memory( |
| 764 | temp_sr, selected_rows_bytes |
| 765 | ) |
| 766 | _seek = f.tell() |
| 767 | |
| 768 | else: |
| 769 | raise NotImplementedError( |
| 770 | f'Only supports load objects from file or BytesIO, but received {type(file_name)}' |
| 771 | ) |
| 772 | |
| 773 | return temp_sr, _seek |
| 774 | |
| 775 | |
| 776 | def _save_binary_var(obj, path): |
no test coverage detected