From 45d3164c00ce9d3e490adca9a2dc779c704c31f0 Mon Sep 17 00:00:00 2001
From: matrixzhou <matrixzhou@google.com>
Date: Wed, 16 Oct 2024 14:58:08 -0700
Subject: [PATCH] NA

PiperOrigin-RevId: 686653252
---
 tensorflow_model_analysis/utils/model_util_test.py | 5 -----
 1 file changed, 5 deletions(-)

diff --git a/tensorflow_model_analysis/utils/model_util_test.py b/tensorflow_model_analysis/utils/model_util_test.py
index 9d66997377..d10984f722 100644
--- 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]