Skip to content

Commit 3744538

Browse files
authored
Bug fix with serialization when walking through operations. (#21532)
Also update error message which was inaccurate.
1 parent 7bf852c commit 3744538

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

keras/src/saving/saving_lib.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1584,7 +1584,7 @@ def get_attr_skipset(obj_type):
15841584

15851585
ref_obj = Operation()
15861586
skipset.update(dir(ref_obj))
1587-
if obj_type == "Layer":
1587+
elif obj_type == "Layer":
15881588
from keras.src.layers.layer import Layer
15891589

15901590
ref_obj = Layer()
@@ -1631,7 +1631,7 @@ def get_attr_skipset(obj_type):
16311631
raise ValueError(
16321632
f"get_attr_skipset got invalid {obj_type=}. "
16331633
"Accepted values for `obj_type` are "
1634-
"['Layer', 'Functional', 'Sequential', 'Metric', "
1634+
"['Operation', 'Layer', 'Functional', 'Sequential', 'Metric', "
16351635
"'Optimizer', 'Loss', 'Cross', 'Feature']"
16361636
)
16371637

0 commit comments

Comments
 (0)