def test_reset_error_count_not_exists(context_with_data: CodeExecutorContext):
  """Test resetting an error count that doesn't exist."""
  context_with_data.reset_error_count("invocationB")  # Should not raise
  assert "invocationB" not in (
      context_with_data._session_state["_code_executor_error_counts"]
  )
Test resetting an error count that doesn't exist.
tests/unittests/code_executors/test_code_executor_context.py