def __init__()

in pipelines/ml_ai_python/ml_ai_pipeline/model_handlers.py [0:0]

• 4 lines of code
• 1 McCabe index (conditional complexity)

  def __init__(self, model_name: str = "gemma_2B"):
    """ Implementation of the ModelHandler interface for Gemma using text as input.

    Example Usage::

      pcoll | RunInference(GemmaModelHandler())

    Args:
      model_name: The Gemma model name. Default is gemma_2B.
    """
    super().__init__()
    self._model_name = model_name
    self._env_vars = {}