Composition: set most_recent_context in execute

- other Components generally set in their execute methods - setting only in Composition.run fails to set when a Composition is nested, because only execute gets called - must still be set in Composition.run in case Composition is empty
PrincetonUniversity · Feb 15, 2025 · 0ee661d · 0ee661d
1 parent 05b8dfb
commit 0ee661d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/psyneulink/core/compositions/composition.py b/psyneulink/core/compositions/composition.py
@@ -12689,6 +12689,8 @@ def execute(
 
             execution_scheduler.get_clock(context)._increment_time(TimeScale.TRIAL)
 
+            self.most_recent_context = context
+
             # Extract result here
             if execution_mode.is_compiled():
                 assert execution_mode & pnlvm.ExecutionMode._PerNode

diff --git a/psyneulink/library/compositions/autodiffcomposition.py b/psyneulink/library/compositions/autodiffcomposition.py
@@ -1525,6 +1525,7 @@ def execute(self,
 
                 scheduler.get_clock(context)._increment_time(TimeScale.TRIAL)
 
+                self.most_recent_context = context
                 return all_output_values
 
         # Call Composition execute in Python mode