Fix 'with_token' type in method device.py/command

Type should be Union[bool, Int]