I want nvitop to expose a one-shot take snapshots(devices: Device Iterable Device None = None, …

I want nvitop to expose a one-shot take snapshots(devices: Device Iterable Device None = None, …: a task in MiMo-V2.6-RL-oss: Agentic RL Environments (MiMo RL release). I want nvitop to expose a one-shot take snapshots(devices: Device Iterable Device None = None, , gpu processes: bool GpuProcess…

The task

I want nvitop to expose a one-shot take_snapshots(devices: Device | Iterable[Device] | None = None, , gpu_processes: bool | GpuProcess | Iterable[GpuProcess] | None = None) -> SnapshotResult function from both nvitop and nvitop.api. It should return a named tuple with .devices…

Part of XiaomiMiMo/MiMo-V2.6-RL-oss.