RSDK-7670 Fix flaking test (viamrobotics#3981)

agreenb · May 21, 2024 · 646fc61 · 646fc61
1 parent 27086ab
commit 646fc61
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/components/base/kinematicbase/execution.go b/components/base/kinematicbase/execution.go
@@ -76,7 +76,7 @@ func (ptgk *ptgBaseKinematics) GoToInputs(ctx context.Context, inputSteps ...[]r
 
 	defer func() {
 		ptgk.inputLock.Lock()
-		ptgk.currentState.currentInputs = zeroInput
+		ptgk.currentState = baseState{currentInputs: zeroInput}
 		ptgk.inputLock.Unlock()
 	}()
 

diff --git a/components/base/kinematicbase/fake_kinematics.go b/components/base/kinematicbase/fake_kinematics.go
@@ -268,9 +268,11 @@ func (fk *fakePTGKinematics) GoToInputs(ctx context.Context, inputSteps ...[]ref
 
 		fk.inputLock.Lock()
 		fk.currentIndex = i
-		fk.currentInput = []referenceframe.Input{inputs[0], inputs[1], inputs[2], inputs[2]}
+		fk.currentInput, err = fk.frame.Interpolate(zeroInput, inputs, 0)
 		fk.inputLock.Unlock()
-
+		if err != nil {
+			return err
+		}
 		finalPose, err := fk.frame.Transform(inputs)
 		if err != nil {
 			return err