attempt to get workspace from attribute of passed argument
@@ -104,6 +104,9 @@ class VehicleBase(ABC):
self._x_hist = []
if workspace:
+ try:
+ self._workspace = workspace.workspace
+ except AttributeError:
self._workspace = base.expand_dims(workspace)
else:
self._workspace = None
