Merge pull request #3209 from PrincetonUniversity/patch/autodiffcompo…

…sition-show-graph-jupyter add return value to show_graph
PrincetonUniversity · Feb 21, 2025 · a2133f8 · a2133f8
2 parents b48c2b6 + bc7f3cc
commit a2133f8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/psyneulink/library/compositions/autodiffcomposition.py b/psyneulink/library/compositions/autodiffcomposition.py
@@ -1755,4 +1755,4 @@ def _get_state_initializer(self, context):
 
     def show_graph(self, *args, **kwargs):
         """Override to use PytorchShowGraph if show_pytorch is True"""
-        self._show_graph.show_graph(*args, **kwargs)
+        return self._show_graph.show_graph(*args, **kwargs)