trying to fix pt.3

PeerDB-io · Dec 20, 2023 · dbe2ad6 · dbe2ad6
1 parent 7d12f41
commit dbe2ad6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/flow/e2e/bigquery/bigquery_helper.go b/flow/e2e/bigquery/bigquery_helper.go
@@ -453,7 +453,7 @@ func (b *BigQueryTestHelper) CreateTable(tableName string, schema *model.QRecord
 	return nil
 }
 
-func (b *BigQueryTestHelper) RunIntQuery(query string) (int64, error) {
+func (b *BigQueryTestHelper) RunIntQuery(query string) (int, error) {
 	recordBatch, err := b.ExecuteAndProcessQuery(query)
 	if err != nil {
 		return 0, fmt.Errorf("could not execute query: %w", err)
@@ -462,5 +462,5 @@ func (b *BigQueryTestHelper) RunIntQuery(query string) (int64, error) {
 		return 0, fmt.Errorf("expected only 1 record, got %d", recordBatch.NumRecords)
 	}
 
-	return recordBatch.Records[0].Entries[0].Value.(int64), nil
+	return int(recordBatch.Records[0].Entries[0].Value.(int64)), nil
 }