class_docstr: Model loader is responsible for loading shared model types.
  
  Attributes:
    construct_fn: A callable which creates the model instance. The callable
      should take no args as input (typically a closure is used to capture
      necessary parameters).
    tags: Optional model tags (e.g. 'serve' for serving or 'eval' for
      EvalSavedModel).
class_docstr_tok: ["Model","loader","is","responsible","for","loading","shared","model","types",".","Attributes",":","construct_fn",":","A","callable","which","creates","the","model","instance",".","The","callable","should","take","no","args","as","input","(","typically","a","closure","is","used","to","capture","necessary","parameters",")",".","tags",":","Optional","model","tags","(","e.g",".","'serve","'","for","serving","or","'eval","'","for","EvalSavedModel",")","."]
docstr: Returns loaded model.
  
  Args:
    model_load_time_callback: Optional callback to track load time.
docstr_tok: ["Returns","loaded","model",".","Args",":","model_load_time_callback",":","Optional","callback","to","track","load","time","."]
code: def load(
      self,
      model_load_time_callback: Optional[Callable[[int], None]] = None,
  ) -> Any:
      """Returns loaded model.
  
      Args:
        model_load_time_callback: Optional callback to track load time.
      """
      if model_load_time_callback:
          construct_fn = self._construct_fn_with_load_time(
              model_load_time_callback
          )
      else:
          construct_fn = self.construct_fn
      return self._shared_handle.acquire(construct_fn)
  
imports: import abc
  import datetime
  import operator
  from typing import Any, Callable, Dict, Iterable, List, MutableMapping, NamedTuple, Optional, TypeVar, Tuple, Union
  from apache_beam.utils import shared
  import numpy
  import six
  import tensorflow
  from tensorflow_model_analysis.proto import metrics_for_slice_pb2
label_desc: Use image node_id 2 for calling the ModelLoader obj's underlying member method code with example usage: obj.load(model_load_time_callback) and returns: self