From 28f8a81edd1727b3567530698b327347f4be7cb6 Mon Sep 17 00:00:00 2001
From: tilakrayal <81610181+tilakrayal@users.noreply.github.com>
Date: Wed, 24 Jan 2024 10:46:00 +0530
Subject: [PATCH] Updated the export_lib.py

In Tensorflow v2.15, tf.types.experimental.GenericFunction has been renamed to tf.types.experimental.PolymorphicFunction
---
 tf_keras/export/export_lib.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tf_keras/export/export_lib.py b/tf_keras/export/export_lib.py
index 88a967cfc..4e44fc753 100644
--- a/tf_keras/export/export_lib.py
+++ b/tf_keras/export/export_lib.py
@@ -244,7 +244,7 @@ def serving_fn(x):
             decorated_fn = tf.function(fn, input_signature=input_signature)
             self._endpoint_signatures[name] = input_signature
         else:
-            if isinstance(fn, tf.types.experimental.GenericFunction):
+            if isinstance(fn, tf.types.experimental.PolymorphicFunction):
                 if not fn._list_all_concrete_functions():
                     raise ValueError(
                         f"The provided tf.function '{fn}' "