File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -428,6 +428,8 @@ def doctest(session: nox.sessions.Session):
428428 "bigframes/display/anywidget.py" ,
429429 "--ignore" ,
430430 "bigframes/bigquery/_operations/ai.py" ,
431+ "--ignore" ,
432+ "bigframes/operations/ai.py" ,
431433 ),
432434 test_folder = "bigframes" ,
433435 check_cov = True ,
Original file line number Diff line number Diff line change @@ -251,12 +251,17 @@ def test_on_event_with_local_execute_result():
251251 import bigframes .core .events
252252 from bigframes .session .executor import LocalExecuteResult
253253
254- local_result = unittest .mock .create_autospec (LocalExecuteResult , instance = True )
254+ # fmt: off
255+ local_result = unittest .mock .create_autospec (
256+ LocalExecuteResult , instance = True
257+ )
258+ # fmt: on
255259 local_result .total_bytes_processed = 1024
256260
257261 event = bigframes .core .events .ExecutionFinished (result = local_result )
262+ envelope = bigframes .core .events .EventEnvelope (event )
258263 execution_metrics = metrics .ExecutionMetrics ()
259- execution_metrics .on_event (event )
264+ execution_metrics .on_event (envelope )
260265
261266 assert execution_metrics .execution_count == 1
262267 assert len (execution_metrics .jobs ) == 1
You can’t perform that action at this time.
0 commit comments