diff --git a/src/beamngpy/api/beamng/camera.py b/src/beamngpy/api/beamng/camera.py index 553c0174..a1d34c90 100644 --- a/src/beamngpy/api/beamng/camera.py +++ b/src/beamngpy/api/beamng/camera.py @@ -80,7 +80,7 @@ def set_player_mode(self, vehicle: str | Vehicle, mode: str, config: StrDict, cu custom_data: Custom data used by the specific camera mode. Defaults to None. """ data: StrDict = dict(type='SetPlayerCameraMode') - data['vid'] = vehicle.vid if isinstance(vehicle, Vehicle) else vehicle + data['vid'] = vehicle if isinstance(vehicle, str) else vehicle.vid data['mode'] = mode data['config'] = config data['customData'] = custom_data @@ -98,7 +98,7 @@ def get_player_modes(self, vehicle: str | Vehicle) -> StrDict: A dictionary mapping camera mode names to configuration options. """ data = dict(type='GetPlayerCameraMode') - data['vid'] = vehicle.vid if isinstance(vehicle, Vehicle) else vehicle + data['vid'] = vehicle if isinstance(vehicle, str) else vehicle.vid resp = self._send(data).recv('PlayerCameraMode') return resp['cameraData']