(data, name)
| 67 | |
| 68 | |
| 69 | def dump_data(data, name): |
| 70 | enable = os.getenv('VERL_ENABLE_TRACKER', '0') == '1' |
| 71 | if not enable: |
| 72 | return |
| 73 | buffer = io.BytesIO() |
| 74 | torch.save(data, buffer) |
| 75 | tracker = get_trajectory_tracker() |
| 76 | ray.get(tracker.dump.remote(buffer, name)) |
| 77 | |
| 78 | |
| 79 | def get_trajectory_tracker(): |
no test coverage detected