Run black 24.1.0

robotpy · Jan 26, 2024 · b0e5f19 · b0e5f19
1 parent 24fea60
commit b0e5f19
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 10 deletions.
diff --git a/MecanumBot/drivetrain.py b/MecanumBot/drivetrain.py
@@ -117,11 +117,13 @@ def drive(
         """Method to drive the robot using joystick info."""
         mecanumDriveWheelSpeeds = self.kinematics.toWheelSpeeds(
             ChassisSpeeds.discretize(
-                ChassisSpeeds.fromFieldRelativeSpeeds(
-                    xSpeed, ySpeed, rot, self.gyro.getRotation2d()
-                )
-                if fieldRelative
-                else ChassisSpeeds(xSpeed, ySpeed, rot),
+                (
+                    ChassisSpeeds.fromFieldRelativeSpeeds(
+                        xSpeed, ySpeed, rot, self.gyro.getRotation2d()
+                    )
+                    if fieldRelative
+                    else ChassisSpeeds(xSpeed, ySpeed, rot)
+                ),
                 periodSeconds,
             )
         )

diff --git a/SwerveBot/drivetrain.py b/SwerveBot/drivetrain.py
@@ -70,11 +70,13 @@ def drive(
         """
         swerveModuleStates = self.kinematics.toSwerveModuleStates(
             wpimath.kinematics.ChassisSpeeds.discretize(
-                wpimath.kinematics.ChassisSpeeds.fromFieldRelativeSpeeds(
-                    xSpeed, ySpeed, rot, self.gyro.getRotation2d()
-                )
-                if fieldRelative
-                else wpimath.kinematics.ChassisSpeeds(xSpeed, ySpeed, rot),
+                (
+                    wpimath.kinematics.ChassisSpeeds.fromFieldRelativeSpeeds(
+                        xSpeed, ySpeed, rot, self.gyro.getRotation2d()
+                    )
+                    if fieldRelative
+                    else wpimath.kinematics.ChassisSpeeds(xSpeed, ySpeed, rot)
+                ),
                 periodSeconds,
             )
         )