Merge pull request #26384 from ricdigi/optimize-point-frame-partial-v…

…elocity physics/vector: Improved speed of Point/Frame partial_velocity methods
sympy · Apr 23, 2024 · ab88a7d · ab88a7d
2 parents 3b6df5e + a0b9430
commit ab88a7d
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 4 deletions.
diff --git a/sympy/physics/vector/frame.py b/sympy/physics/vector/frame.py
@@ -1558,8 +1558,10 @@ def partial_velocity(self, frame, *gen_speeds):
 
         """
 
-        partials = [self.ang_vel_in(frame).diff(speed, frame, var_in_dcm=False)
-                    for speed in gen_speeds]
+        from sympy.physics.vector.functions import partial_velocity
+
+        vel = self.ang_vel_in(frame)
+        partials = partial_velocity([vel], gen_speeds, frame)[0]
 
         if len(partials) == 1:
             return partials[0]

diff --git a/sympy/physics/vector/point.py b/sympy/physics/vector/point.py
@@ -623,8 +623,11 @@ def partial_velocity(self, frame, *gen_speeds):
         (N.x, A.y)
 
         """
-        partials = [self.vel(frame).diff(speed, frame, var_in_dcm=False) for
-                    speed in gen_speeds]
+
+        from sympy.physics.vector.functions import partial_velocity
+
+        vel = self.vel(frame)
+        partials = partial_velocity([vel], gen_speeds, frame)[0]
 
         if len(partials) == 1:
             return partials[0]