code: def __init__(self, context: keras.TFKerasTrialContext) -> None:
      self.context = context
  
file_path: determined/examples/computer_vision/iris_tf_keras/model_def.py
imports: from typing import List
  import pandas
  import tensorflow
  from tensorflow.keras.layers import Dense, Input
  from tensorflow.keras.losses import categorical_crossentropy
  from tensorflow.keras.metrics import categorical_accuracy
  from tensorflow.keras.models import Model
  from tensorflow.keras.optimizers.legacy import RMSprop
  from tensorflow.keras.utils import to_categorical
  from determined import keras
label_desc: Use image node_id 1 to create a new IrisTrial object from inherited base classes: keras with example: obj = IrisTrial(context)