fix: don't raise UnknownLocationWarning for US or EU multi-regions (#653)

tswast · web-flow · commit 8e4616b896f4 · 2024-05-02T15:39:12.000-05:00
diff --git a/bigframes/constants.py b/bigframes/constants.py
@@ -31,6 +31,7 @@
 # https://cloud.google.com/bigquery/docs/locations
 ALL_BIGQUERY_LOCATIONS = frozenset(
     {
+        # regions
         "us-east5",
         "us-south1",
         "us-central1",
@@ -74,6 +75,9 @@
         "me-central1",
         "me-west1",
         "africa-south1",
+        # multi-regions
+        "US",
+        "EU",
     }
 )
 

Original file line number	Diff line number	Diff line change
`@@ -31,6 +31,7 @@`
`31`	`31`	`# https://cloud.google.com/bigquery/docs/locations`
`32`	`32`	`ALL_BIGQUERY_LOCATIONS = frozenset(`
`33`	`33`	`{`
	`34`	`+ # regions`
`34`	`35`	`"us-east5",`
`35`	`36`	`"us-south1",`
`36`	`37`	`"us-central1",`
`@@ -74,6 +75,9 @@`
`74`	`75`	`"me-central1",`
`75`	`76`	`"me-west1",`
`76`	`77`	`"africa-south1",`
	`78`	`+ # multi-regions`
	`79`	`+ "US",`
	`80`	`+ "EU",`
`77`	`81`	`}`
`78`	`82`	`)`
`79`	`83`