Pass location to the BigQuery request (flyteorg#365)

Signed-off-by: Kevin Su <[email protected]>
eapolinario · Jun 27, 2023 · 3f702bb · 3f702bb
1 parent 6922a96
commit 3f702bb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/flyteplugins/go/tasks/plugins/webapi/bigquery/plugin.go b/flyteplugins/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