no-op

PiperOrigin-RevId: 549352767
tensorflow · Jul 19, 2023 · 6b5d328 · 6b5d328
1 parent 98fa7ec
commit 6b5d328
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tensorflow_model_analysis/api/model_eval_lib.py b/tensorflow_model_analysis/api/model_eval_lib.py
@@ -412,8 +412,11 @@ def default_eval_shared_model(
       with *some* prediction extractors.
   """
   if not eval_config:
+    # Default to tfma eval model unless eval
     is_baseline = False
-    if tags and tags == [tf.saved_model.SERVING]:
+    if tags and eval_constants.EVAL_TAG in tags:
+      model_type = constants.TFMA_EVAL
+    elif tags and tf.saved_model.SERVING in tags:
       model_type = constants.TF_ESTIMATOR
     else:
       model_type = constants.TFMA_EVAL