NA

PiperOrigin-RevId: 686653252
tensorflow · Oct 16, 2024 · 45d3164 · 45d3164
1 parent fad48c8
commit 45d3164
Showing 1 changed file with 0 additions and 5 deletions.
diff --git a/tensorflow_model_analysis/utils/model_util_test.py b/tensorflow_model_analysis/utils/model_util_test.py
@@ -750,11 +750,6 @@ def testInputSpecsToTensorRepresentationsRaisesWithUnknownDims(self):
           }
       }, ['unknown', 'custom_single_output', 'custom_multi_output'
          ], True, True, 3),
-      ('custom_attribute', True, {
-          constants.TRANSFORMED_FEATURES_KEY: {
-              '': ['custom_attribute'],
-          }
-      }, None, True, True, 3),
       ('keras_no_schema', True, {
           constants.PREDICTIONS_KEY: {
               '': [None]