Using utils test_util for unbatch_extractor_test

PiperOrigin-RevId: 685795373

tensorflow_model_analysis/extractors/unbatch_extractor_test.py

@@ -19,13 +19,13 @@
 import tensorflow as tf
 from tensorflow_model_analysis import constants
 from tensorflow_model_analysis.api import model_eval_lib
-from tensorflow_model_analysis.eval_saved_model import testutil
 from tensorflow_model_analysis.extractors import example_weights_extractor
 from tensorflow_model_analysis.extractors import features_extractor
 from tensorflow_model_analysis.extractors import labels_extractor
 from tensorflow_model_analysis.extractors import materialized_predictions_extractor
 from tensorflow_model_analysis.extractors import unbatch_extractor
 from tensorflow_model_analysis.proto import config_pb2
+from tensorflow_model_analysis.utils import test_util as testutil
 from tfx_bsl.tfxio import test_util
 
 from google.protobuf import text_format