fix test

sycai · sycai · commit 1769df0f10c2 · 2025-10-21T20:23:16.000Z
diff --git a/bigframes/session/dry_runs.py b/bigframes/session/dry_runs.py
@@ -160,7 +160,10 @@ def get_query_stats(
     )
 
     result = pandas.Series(values, index=index)
-    result["totalBytesProcessed"] = int(result["totalBytesProcessed"])
+    if result["totalBytesProcessed"] is None:
+        result["totalBytesProcessed"] = 0
+    else:
+        result["totalBytesProcessed"] = int(result["totalBytesProcessed"])
 
     return result
 

Original file line number	Diff line number	Diff line change
`@@ -160,7 +160,10 @@ def get_query_stats(`
`160`	`160`	`)`
`161`	`161`
`162`	`162`	`result = pandas.Series(values, index=index)`
`163`		`- result["totalBytesProcessed"] = int(result["totalBytesProcessed"])`
	`163`	`+ if result["totalBytesProcessed"] is None:`
	`164`	`+ result["totalBytesProcessed"] = 0`
	`165`	`+ else:`
	`166`	`+ result["totalBytesProcessed"] = int(result["totalBytesProcessed"])`
`164`	`167`
`165`	`168`	`return result`
`166`	`169`