Update some comments

tensorflow · Nov 27, 2024 · 3550744 · 3550744
1 parent 157ff40
commit 3550744
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/tfx/examples/penguin/experimental/penguin_pipeline_sklearn_local.py b/tfx/examples/penguin/experimental/penguin_pipeline_sklearn_local.py
@@ -117,6 +117,8 @@ def _create_pipeline(
           filesystem=tfx.proto.PushDestination.Filesystem(
               base_directory=serving_model_dir)))
 
+  # Note: Because TFMA 0.47.0 doesn't support custom model evaluation,
+  # the evaluator step is ruled out here.
   return tfx.dsl.Pipeline(
       pipeline_name=pipeline_name,
       pipeline_root=pipeline_root,

diff --git a/tfx/examples/penguin/penguin_pipeline_local_e2e_test.py b/tfx/examples/penguin/penguin_pipeline_local_e2e_test.py
@@ -226,7 +226,8 @@ def testPenguinPipelineLocalWithTuner(self):
 
   @parameterized.parameters(('keras',), ('flax_experimental',),
                             ('tfdf_experimental',))
-  @pytest.mark.xfail(run=False, reason="Keras model is not working with bulk inference now. Need to fix")
+  @pytest.mark.xfail(run=False,
+                     reason="Exported Keras model with TF 1.16 is not working with bulk inference currently. Needs to be fixed.")
   def testPenguinPipelineLocalWithBulkInferrer(self, model_framework):
     if model_framework == 'tfdf_experimental':
       # Skip if TFDF is not available or incompatible.