Skip to content

Commit 84ebef2

Browse files
committed
Add coordinate system type as argument anywhere that coordinates are provided
1 parent 949ff0d commit 84ebef2

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

lgsvl/dreamview/dreamview.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -261,15 +261,15 @@ def reconnect(self):
261261
self.ws = create_connection(self.url)
262262
return
263263

264-
def enable_apollo(self, dest_x, dest_z, modules):
264+
def enable_apollo(self, dest_x, dest_z, modules, coord_type=CoordType.Unity):
265265
"""
266266
Enables a list of modules and then sets the destination
267267
"""
268268
for mod in modules:
269269
log.info("Starting {} module...".format(mod))
270270
self.enable_module(mod)
271271

272-
self.set_destination(dest_x, dest_z)
272+
self.set_destination(dest_x, dest_z, coord_type=coord_type)
273273

274274
def disable_apollo(self):
275275
"""
@@ -290,7 +290,7 @@ def check_module_status(self, modules):
290290
"Warning: Apollo module {} is not running!!!".format(module)
291291
)
292292

293-
def setup_apollo(self, dest_x, dest_z, modules, default_timeout=60.0):
293+
def setup_apollo(self, dest_x, dest_z, modules, default_timeout=60.0, coord_type=CoordType.Unity):
294294
"""
295295
Starts a list of Apollo modules and sets the destination. Will wait for Control module to send a message before returning.
296296
Control sending a message indicates that all modules are working and Apollo is ready to continue.
@@ -302,7 +302,7 @@ def setup_apollo(self, dest_x, dest_z, modules, default_timeout=60.0):
302302
if not all(mod_status[mod] for mod in modules):
303303
self.disable_apollo()
304304

305-
self.enable_apollo(dest_x, dest_z, modules)
305+
self.enable_apollo(dest_x, dest_z, modules, coord_type=coord_type)
306306
self.ego.is_control_received = False
307307

308308
def on_control_received(agent, kind, context):

0 commit comments

Comments
 (0)