Skip to content

Commit f153425

Browse files
Olivia-liufacebook-github-bot
authored andcommitted
Fix exir deserialization logic (#2857)
Summary: Looks like a recent str -> TensorArgument change in diff D54855251 wasn't very thorough. ETRecord deserialization failed because of it. Differential Revision: D55776877
1 parent f64130e commit f153425

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

exir/serde/export_serialize.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1503,7 +1503,7 @@ def deserialize_optional_tensor_args(a):
15031503
if a.type == "as_none":
15041504
return None
15051505
elif a.type == "as_tensor":
1506-
return self.serialized_name_to_node[a.value]
1506+
return self.serialized_name_to_node[a.value.name]
15071507
else:
15081508
raise SerializeError(f"Unhandled argument {inp}")
15091509

0 commit comments

Comments
 (0)