From 2f62caa4e5fb953c7afb3144653b8c2f052aaf7e Mon Sep 17 00:00:00 2001
From: Yaohui Liu <AsakusaRinne@gmail.com>
Date: Tue, 18 Apr 2023 12:37:35 +0800
Subject: [PATCH] Revise some details.

---
 src/TensorFlowNET.Keras/Saving/SavedModel/load.cs              | 3 ++-
 .../SaveModel/SequentialModelLoad.cs                           | 2 +-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/TensorFlowNET.Keras/Saving/SavedModel/load.cs b/src/TensorFlowNET.Keras/Saving/SavedModel/load.cs
index abb2012f..362464d1 100644
--- a/src/TensorFlowNET.Keras/Saving/SavedModel/load.cs
+++ b/src/TensorFlowNET.Keras/Saving/SavedModel/load.cs
@@ -56,7 +56,8 @@ namespace Tensorflow.Keras.Saving.SavedModel
             }
             else
             {
-                throw new NotImplementedException("Not implemented, please submit an issue to https://github.com/SciSharp/TensorFlow.NET/issues.");
+                throw new NotImplementedException("SavedModel saved prior to TF 2.5 detected when loading Keras model, please" +
+                    " use higher version or submit an issue to https://github.com/SciSharp/TensorFlow.NET/issues. to let us know you need it.");
             }
 
             if (metadata.Nodes is null || metadata.Nodes.Count == 0)
diff --git a/test/TensorFlowNET.Keras.UnitTest/SaveModel/SequentialModelLoad.cs b/test/TensorFlowNET.Keras.UnitTest/SaveModel/SequentialModelLoad.cs
index 51962830..f91f1fe7 100644
--- a/test/TensorFlowNET.Keras.UnitTest/SaveModel/SequentialModelLoad.cs
+++ b/test/TensorFlowNET.Keras.UnitTest/SaveModel/SequentialModelLoad.cs
@@ -59,7 +59,7 @@ public class SequentialModelLoad
     }
 
     [TestMethod]
-    public void Temp()
+    public void ModelWithSelfDefinedModule()
     {
         var model = tf.keras.models.load_model(@"Assets/python_func_model");
         model.summary();