Small fix for v.3.12

Thoughtseize1 · Thoughtseize1 · commit e0fbccf59508 · 2024-11-18T18:12:41.000+01:00
diff --git a/tpu/test_tpu.py b/tpu/test_tpu.py
@@ -74,7 +74,7 @@ def test_creating_tpu(mock_tpu_client: MagicMock, operation: MagicMock) -> None:
 
 def test_delete_tpu(mock_tpu_client: MagicMock) -> None:
     delete_tpu.delete_cloud_tpu(PROJECT_ID, ZONE, TPU_NAME)
-    mock_tpu_client.delete_node.called_once()
+    mock_tpu_client.delete_node.assert_called_once()
 
 
 def test_creating_with_startup_script(
