From 9a7b81c2e717cb92075bb5300b32a668040fe4aa Mon Sep 17 00:00:00 2001 From: Christian Orr Date: Sat, 22 Apr 2023 09:15:05 +0200 Subject: [PATCH] Update basic.py Printing the updated nested graph --- jraph/examples/basic.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/jraph/examples/basic.py b/jraph/examples/basic.py index 66f335c..bf5d73b 100644 --- a/jraph/examples/basic.py +++ b/jraph/examples/basic.py @@ -194,7 +194,7 @@ def update_globals_fn( logging.info("Updated graph from single graph %r", updated_graph) updated_graph = network(nested_graph) - logging.info("Updated graph from nested graph %r", nested_graph) + logging.info("Updated graph from nested graph %r", updated_graph) updated_graph = network(implicitly_batched_graph) logging.info("Updated graph from implicitly batched graph %r", updated_graph)