| 1867 | return dict(record.tensors) |
| 1868 | |
| 1869 | def release(self, update_id: str) -> None: |
| 1870 | super().release(update_id) |
| 1871 | segments = self._shared_memory_segments.pop(update_id, {}) |
| 1872 | should_unlink = update_id in self._owned_shared_memory_update_ids |
| 1873 | self._owned_shared_memory_update_ids.discard(update_id) |
| 1874 | for segment in segments.values(): |
| 1875 | try: |
| 1876 | segment.close() |
| 1877 | finally: |
| 1878 | if should_unlink: |
| 1879 | try: |
| 1880 | segment.unlink() |
| 1881 | except FileNotFoundError: |
| 1882 | pass |
| 1883 | |
| 1884 | def _attach_shared_memory_manifest(self, manifest: WeightUpdateManifest) -> _LocalUpdateRecord: |
| 1885 | if manifest.transport != self.transport: |