make sure JSON is callable

googleapis · Mar 10, 2025 · df87a70 · df87a70
1 parent 7b225d7
commit df87a70
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/google/cloud/bigquery/_pyarrow_helpers.py b/google/cloud/bigquery/_pyarrow_helpers.py
@@ -87,7 +87,7 @@ def pyarrow_timestamp():
         "GEOGRAPHY": pyarrow.string,
         "INT64": pyarrow.int64,
         "INTEGER": pyarrow.int64,
-        "JSON": json_arrow_type,
+        "JSON": lambda: json_arrow_type,
         "NUMERIC": pyarrow_numeric,
         "STRING": pyarrow.string,
         "TIME": pyarrow_time,

diff --git a/tests/unit/test__pyarrow_helpers.py b/tests/unit/test__pyarrow_helpers.py
@@ -27,11 +27,11 @@ def module_under_test():
 
 def test_bq_to_arrow_scalars(module_under_test):
     assert (
-        module_under_test.bq_to_arrow_scalars("BIGNUMERIC")
-        == module_under_test.pyarrow_bignumeric
+        module_under_test.bq_to_arrow_scalars("BIGNUMERIC")()
+        == module_under_test.pyarrow_bignumeric()
     )
     assert (
-        module_under_test.bq_to_arrow_scalars("JSON")
+        module_under_test.bq_to_arrow_scalars("JSON")()
         == module_under_test.json_arrow_type
     )
     assert module_under_test.bq_to_arrow_scalars("UNKNOWN_TYPE") is None