code: def main():
      wandb.init(
          name=pathlib.Path(__file__).stem,
          settings=wandb.Settings(
              _stats_sample_rate_seconds=1,
              _stats_samples_to_average=1,
          ),
      )
  
      model = tf.keras.models.Sequential()
      model.add(
          tf.keras.layers.Conv2D(
              3, 3, activation="relu", input_shape=(28, 28, 1)
          )
      )
      model.add(tf.keras.layers.Flatten())
      model.add(tf.keras.layers.Dense(10, activation="softmax"))
      model.compile(
          loss="sparse_categorical_crossentropy",
          optimizer="sgd",
          metrics=["accuracy"],
      )
  
      model.fit(
          np.ones((10, 28, 28, 1)),
          np.ones((10,)),
          epochs=7,
          validation_split=0.2,
          callbacks=[WandbCallback()],
      )
  
imports: import pathlib
  import keras
  import numpy
  import tensorflow
  import wandb
  from wandb.keras import WandbCallback
label_desc: Use image node_id 1 for calling a global function with example usage: main() without return types