code: def __init__(self, num_classes=10):
      super().__init__()
      self.conv = tf.keras.layers.Conv2D(
          3, 3, activation="relu", input_shape=(28, 28, 1)
      )
      self.flatten = tf.keras.layers.Flatten()
      self.classifier = tf.keras.layers.Dense(10, activation="softmax")
  
file_path: wandb/tests/functional_tests/t0_main/keras/test_keras_subclassed_model.py
imports: import os
  import numpy
  import tensorflow
  import wandb
  from wandb.keras import WandbCallback
label_desc: Use image node_id 1 to create a new DummyModel object from inherited base classes: tf.keras with example: obj = DummyModel(num_classes)