bug fix
This commit is contained in:
@@ -65,6 +65,6 @@ class ReactGraph(GraphBase):
|
|||||||
msgs_list = jax.tree.leaves(out)
|
msgs_list = jax.tree.leaves(out)
|
||||||
|
|
||||||
if as_raw:
|
if as_raw:
|
||||||
return out
|
return msgs_list
|
||||||
else:
|
else:
|
||||||
return msgs_list[-1].content
|
return msgs_list[-1].content
|
||||||
|
|||||||
Reference in New Issue
Block a user