code: def log_after_retry(name):
      return lambda retry_state: logger.info(
          "Retrying TF serving connection",
          name=name,
          attempt_number=retry_state.attempt_number,
          wait_time=retry_state.outcome_timestamp
          - retry_state.start_time,
      )
  
imports: import abc
  import json
  import os
  from typing import Any, Dict, List, Optional, Tuple, Type
  import aiohttp
  import requests
  from structlog import get_logger
  from tenacity import Retrying, retry_if_exception, stop_after_attempt, wait_random_exponential
  from modelkit.core.model import AsyncModel, Model
  from modelkit.core.types import ItemType, ReturnType
  from modelkit.utils.serialization import safe_np_dump
label_desc: Use image node_id 1 for calling a global function with example usage: log_after_retry(name) and returns: function