diff --git a/omnigibson/examples/robots/import_custom_robot.py b/omnigibson/examples/robots/import_custom_robot.py index de50cbcae..6a33887bd 100644 --- a/omnigibson/examples/robots/import_custom_robot.py +++ b/omnigibson/examples/robots/import_custom_robot.py @@ -2,6 +2,7 @@ Helper script to download OmniGibson dataset and assets. """ +import xml.etree.ElementTree as ET from copy import deepcopy from pathlib import Path @@ -9,7 +10,6 @@ import torch as th import yaml from addict import Dict -import xml.etree.ElementTree as ET import omnigibson as og import omnigibson.lazy as lazy @@ -17,9 +17,9 @@ from omnigibson.macros import gm from omnigibson.utils.asset_conversion_utils import ( _add_xform_properties, + _space_string_to_tensor, find_all_prim_children_with_type, import_og_asset_from_urdf, - _space_string_to_tensor, ) from omnigibson.utils.python_utils import assert_valid_key from omnigibson.utils.usd_utils import create_joint, create_primitive_mesh diff --git a/tests/test_object_states.py b/tests/test_object_states.py index 3bc4fe256..82cdd1816 100644 --- a/tests/test_object_states.py +++ b/tests/test_object_states.py @@ -664,7 +664,9 @@ def test_toggled_on(env): stove = env.scene.object_registry("name", "stove") robot = env.robots[0] - stove.set_position_orientation([1.505, 0.3, 0.443], T.euler2quat(th.tensor([0, 0, -math.pi / 2.0], dtype=th.float32))) + stove.set_position_orientation( + [1.505, 0.3, 0.443], T.euler2quat(th.tensor([0, 0, -math.pi / 2.0], dtype=th.float32)) + ) robot.set_position_orientation(position=[0.0, 0.38, 0.0], orientation=[0, 0, 0, 1]) assert not stove.states[ToggledOn].get_value()