Merge pull request #699 from etiennebeaulac/fix-differentialdrive-docs

Fix DifferentialDrive docs examples
robotpy · Jan 26, 2022 · a056891 · a056891
2 parents 94ace25 + f44a66f
commit a056891
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/gen/drive/DifferentialDrive.yml b/gen/drive/DifferentialDrive.yml
@@ -41,6 +41,8 @@ classes:
       :class:`MotorControllerGroup` instances as follows.
 
       Four motor drivetrain::
+        import wpilib
+        import wpilib.drive
 
         class Robot(wpilib.TimedRobot):
             def robotInit(self):
@@ -52,9 +54,11 @@ classes:
                 self.rear_right = wpilib.PWMVictorSPX(4)
                 self.right = wpilib.MotorControllerGroup(self.front_right, self.rear_right)
 
-                self.drive = wpilib.DifferentialDrive(self.left, self.right)
+                self.drive = wpilib.drive.DifferentialDrive(self.left, self.right)
 
       Six motor drivetrain::
+        import wpilib
+        import wpilib.drive
 
         class Robot(wpilib.TimedRobot):
             def robotInit(self):
@@ -68,7 +72,7 @@ classes:
                 self.rear_right = wpilib.PWMVictorSPX(6)
                 self.right = wpilib.MotorControllerGroup(self.front_right, self.mid_right, self.rear_right)
 
-                self.drive = wpilib.DifferentialDrive(self.left, self.right)
+                self.drive = wpilib.drive.DifferentialDrive(self.left, self.right)
 
       A differential drive robot has left and right wheels separated by an
       arbitrary width.