From 50af744a974f7ba3bc6ad1dbb406a4c007862628 Mon Sep 17 00:00:00 2001
From: Shreyas Goenka <shreyas.goenka@databricks.com>
Date: Mon, 18 Nov 2024 16:32:41 +0100
Subject: [PATCH] fix test

---
 bundle/run/pipeline_test.go | 2 --
 1 file changed, 2 deletions(-)

diff --git a/bundle/run/pipeline_test.go b/bundle/run/pipeline_test.go
index e4608061..821e24d4 100644
--- a/bundle/run/pipeline_test.go
+++ b/bundle/run/pipeline_test.go
@@ -90,8 +90,6 @@ func TestPipelineRunnerRestart(t *testing.T) {
 		PipelineId: "123",
 	}).Return(mockWait, nil)
 
-	pipelineApi.EXPECT().GetByPipelineId(mock.Anything, "123").Return(&pipelines.GetPipelineResponse{}, nil)
-
 	// Mock runner starting a new update
 	pipelineApi.EXPECT().StartUpdate(mock.Anything, pipelines.StartUpdate{
 		PipelineId: "123",