Fix TF saving issue

fchollet · fchollet · commit 983635ff9edf · 2023-06-11T22:36:49.000-07:00
diff --git a/keras_core/saving/saving_lib.py b/keras_core/saving/saving_lib.py
@@ -49,6 +49,14 @@
         "variables",
         "non_trainable_variables",
         "trainable_variables",
+        # TF trackable attrs
+        "_unconditional_checkpoint_dependencies",
+        "_unconditional_dependency_names",
+        "_checkpoint_dependencies",
+        "_deferred_dependencies",
+        "_deserialization_dependencies",
+        "_lookup_dependency",
+        "_self_unconditional_dependency_names",
     }
 )
 

Original file line number	Diff line number	Diff line change
`@@ -49,6 +49,14 @@`
`49`	`49`	`"variables",`
`50`	`50`	`"non_trainable_variables",`
`51`	`51`	`"trainable_variables",`
	`52`	`+ # TF trackable attrs`
	`53`	`+ "_unconditional_checkpoint_dependencies",`
	`54`	`+ "_unconditional_dependency_names",`
	`55`	`+ "_checkpoint_dependencies",`
	`56`	`+ "_deferred_dependencies",`
	`57`	`+ "_deserialization_dependencies",`
	`58`	`+ "_lookup_dependency",`
	`59`	`+ "_self_unconditional_dependency_names",`
`52`	`60`	`}`
`53`	`61`	`)`
`54`	`62`