docstr: Tests that the sklearn extractor is used when creating extracts.
docstr_tok: ["Tests","that","the","sklearn","extractor","is","used","when","creating","extracts","."]
code: def test_custom_extractors(self):
      """Tests that the sklearn extractor is used when creating extracts."""
      extractors = sklearn_predict_extractor.custom_extractors(
          self._eval_shared_model,
          self._eval_config,
          self._tensor_adapter_config,
      )
      self.assertLen(extractors, 6)
      self.assertIn(
          "SklearnPredict",
          [extractor.stage_name for extractor in extractors],
      )
  
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 5 for calling the SklearnPredictExtractorTest obj's underlying member method code with example usage: obj.test_custom_extractors() without return types