Various fixes in mujoco/usd

PiperOrigin-RevId: 653610253
Change-Id: Ibe57c19c6a7326266f51c5cda39cd564bd456c54
This commit is contained in:
Alessio Quaglino
2024-07-18 07:19:29 -07:00
committed by Copybara-Service
parent ffb50b7b67
commit 384ed8097a
5 changed files with 23 additions and 9 deletions
+5 -3
View File
@@ -16,13 +16,15 @@
import abc
import collections
from typing import Optional
from typing import Optional, Dict, Any
import mujoco
import mujoco.usd.shapes as shapes_component
import mujoco.usd.utils as utils_component
import numpy as np
# TODO: b/288149332 - Remove once USD Python Binding works well with pytype.
# pytype: disable=module-attr
from pxr import Gf
from pxr import Sdf
from pxr import Usd
@@ -310,7 +312,7 @@ class USDPrimitiveMesh(USDObject):
def __init__(
self,
mesh_config: dict[any, any],
mesh_config: Dict[Any, Any],
stage: Usd.Stage,
geom: mujoco.MjvGeom,
obj_name: str,
@@ -379,7 +381,7 @@ class USDTendon(USDObject):
def __init__(
self,
mesh_config: dict[any, any],
mesh_config: Dict[Any, Any],
stage: Usd.Stage,
geom: mujoco.MjvGeom,
obj_name: str,