Pass location to the BigQuery request (#365)

Signed-off-by: Kevin Su <[email protected]>
flyteorg · Jun 27, 2023 · 1d0f3e9 · 1d0f3e9
1 parent dfdf6f9
commit 1d0f3e9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/go/tasks/plugins/webapi/bigquery/plugin.go b/go/tasks/plugins/webapi/bigquery/plugin.go
@@ -220,7 +220,7 @@ func (p Plugin) getImpl(ctx context.Context, taskCtx webapi.GetContext) (wrapper
 		return nil, pluginErrors.Wrapf(pluginErrors.RuntimeFailure, err, "unable to get client")
 	}
 
-	job, err := client.Jobs.Get(resourceMeta.JobReference.ProjectId, resourceMeta.JobReference.JobId).Do()
+	job, err := client.Jobs.Get(resourceMeta.JobReference.ProjectId, resourceMeta.JobReference.JobId).Location(resourceMeta.JobReference.Location).Do()
 
 	if err != nil {
 		err := pluginErrors.Wrapf(
@@ -256,7 +256,7 @@ func (p Plugin) Delete(ctx context.Context, taskCtx webapi.DeleteContext) error
 		return err
 	}
 
-	_, err = client.Jobs.Cancel(resourceMeta.JobReference.ProjectId, resourceMeta.JobReference.JobId).Do()
+	_, err = client.Jobs.Cancel(resourceMeta.JobReference.ProjectId, resourceMeta.JobReference.JobId).Location(resourceMeta.JobReference.Location).Do()
 
 	if err != nil {
 		return err