diff --git a/jedi/tests/test_get_signature.py b/jedi/tests/test_get_signature.py index 7f82bc3..804f55a 100644 --- a/jedi/tests/test_get_signature.py +++ b/jedi/tests/test_get_signature.py @@ -149,6 +149,7 @@ CONSTRUCTOR_PARAMS = [ "wheel_diameter: Number", "axle_track: Number", "positive_direction: Direction=Direction.CLOCKWISE", + "use_gyro: bool=False", ] ], ), diff --git a/src/pybricks/robotics.py b/src/pybricks/robotics.py index adc543c..c496971 100644 --- a/src/pybricks/robotics.py +++ b/src/pybricks/robotics.py @@ -59,8 +59,9 @@ class DriveBase: wheel_diameter: Number, axle_track: Number, positive_direction: Direction = Direction.CLOCKWISE, + use_gyro: bool = False, ): - """DriveBase(left_motor, right_motor, wheel_diameter, axle_track, positive_direction=Direction.CLOCKWISE) + """DriveBase(left_motor, right_motor, wheel_diameter, axle_track, positive_direction=Direction.CLOCKWISE, use_gyro=False) Arguments: left_motor (Motor): @@ -73,6 +74,12 @@ class DriveBase: positive_direction (Direction): Which direction the drive base should turn when you give a positive turn rate or turn angle, viewed from the top. + use_gyro (bool): Choose ``True`` to use the internal gyrosope for + more accurate turns. This can only be used when + the ``positive_direction`` is set to counterclockwise. This is + only supported on hubs with an inertial measurement unit (IMU). + If you choose ``False`` (the default), turns are measured + using the wheel angles. """ def drive(self, speed: Number, turn_rate: Number) -> None: