File size: 812 Bytes
2385631 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 | """RoboMME package initialization."""
from __future__ import annotations
def _patch_maniskill_pci_render_backend() -> None:
"""Allow ManiSkill to pass PCI-style Vulkan device strings through intact."""
try:
from mani_skill.envs.utils.system import backend as ms_backend
except Exception:
return
if getattr(ms_backend, "_robomme_pci_backend_patch", False):
return
original = ms_backend.parse_backend_device_id
def patched_parse_backend_device_id(backend: str):
if isinstance(backend, str) and backend.startswith("pci:"):
return backend, None
return original(backend)
ms_backend.parse_backend_device_id = patched_parse_backend_device_id
ms_backend._robomme_pci_backend_patch = True
_patch_maniskill_pci_render_backend()
|