code: def test_retry_policy(exception, exc_args, policy):
      SHORT_RETRY_POLICY = copy.deepcopy(policy)
      SHORT_RETRY_POLICY["stop"] = stop_after_attempt(2)
      k = 0
  
      @retry(**SHORT_RETRY_POLICY)
      def some_function():
          nonlocal k
          k += 1
          raise exception(*exc_args)
  
      with pytest.raises(exception):
          some_function()
      assert k == 2
  
file_path: modelkit/tests/assets/test_retry.py
imports: import copy
  import botocore
  import botocore.exceptions
  import google.api_core
  import pytest
  import requests
  from tenacity import retry, stop_after_attempt
  from modelkit.assets.drivers.retry import retry_policy
  from modelkit.assets.errors import ObjectDoesNotExistError
label_desc: Use image node_id 1 for calling a global function with example usage: test_retry_policy(exception, exc_args, policy) without return types