code: def test_keras_image_bad_data():
      import numpy as np
  
      K.clear_session()
      model = Sequential()
      model.add(Flatten(input_shape=(10, 10, 3)))
      model.add(Dense(1, activation="sigmoid"))
      model.compile(
          optimizer="adam",
          loss="binary_crossentropy",
          metrics=["accuracy"],
      )
  
      data = np.random.randint(255, size=(100, 10, 10, 3))
      labels = np.random.randint(2, size=(100, 1))
  
      with pytest.raises(ValueError):
          model.fit(
              data,
              labels,
              epochs=2,
              batch_size=36,
              validation_data=(data.reshape(10), labels),
              callbacks=[
                  WandbCallback(data_type="image"),
              ],
          )
  
file_path: wandb/tests/pytest_tests/unit_tests/test_keras.py
imports: import sys
  import pytest
  import wandb
  from keras import backend
  from keras.layers import LSTM, Concatenate, Dense, Embedding, Flatten, Input
  from keras.models import Model, Sequential
  from wandb.keras import WandbCallback
label_desc: Use image node_id 2 for calling a global function with example usage: test_keras_image_bad_data() without return types