code: def __init__(self, w, b):
      super().__init__()
      self.w = w
      self.b = b
  
file_path: mlflow/tests/tensorflow/test_tensorflow2_core_model_export.py
imports: import collections
  import os
  from unittest import mock
  import numpy
  import pytest
  import tensorflow
  import mlflow.tensorflow
  from mlflow.models import Model, infer_signature
label_desc: Use image node_id 1 to create a new ToyModel object from inherited base classes: tf with example: obj = ToyModel(w, b)