code: def setUp(self):
      super().setUp()
      self._eval_export_dir = os.path.join(
          self._getTempDir(), "eval_export"
      )
      self._create_sklearn_model(self._eval_export_dir)
      self._eval_config = tfma.EvalConfig(
          model_specs=[tfma.ModelSpec()]
      )
      self._eval_shared_model = (
          sklearn_predict_extractor.custom_eval_shared_model(
              eval_saved_model_path=self._eval_export_dir,
              model_name=None,
              eval_config=self._eval_config,
          )
      )
      self._schema = text_format.Parse(
          """
          feature {
            name: "age"
            type: FLOAT
          }
          feature {
            name: "language"
            type: FLOAT
          }
          feature {
            name: "label"
            type: INT
          }
          """,
          schema_pb2.Schema(),
      )
      self._tfx_io = test_util.InMemoryTFExampleRecord(
          schema=self._schema,
          raw_record_column_name=tfma.ARROW_INPUT_COLUMN,
      )
      self._tensor_adapter_config = tensor_adapter.TensorAdapterConfig(
          arrow_schema=self._tfx_io.ArrowSchema(),
          tensor_representations=self._tfx_io.TensorRepresentations(),
      )
      self._examples = [
          self._makeExample(age=3.0, language=1.0, label=1),
          self._makeExample(age=3.0, language=0.0, label=0),
          self._makeExample(age=4.0, language=1.0, label=1),
          self._makeExample(age=5.0, language=0.0, label=0),
      ]
  
file_path: tfx/tfx/examples/penguin/experimental/sklearn_predict_extractor_test.py
imports: import os
  import pickle
  import apache_beam
  from apache_beam.testing import util
  from sklearn import neural_network
  import tensorflow_model_analysis
  from tfx.examples.penguin.experimental import sklearn_predict_extractor
  from tfx_bsl.tfxio import tensor_adapter
  from tfx_bsl.tfxio import test_util
  from google.protobuf import text_format
  from tensorflow_metadata.proto.v0 import schema_pb2
label_desc: Use image node_id 1 for calling the SklearnPredictExtractorTest obj's underlying member method code with example usage: obj.setUp() without return types