Composition: _delete_contexts: fix crash if given execution_id (#2103)

PrincetonUniversity · Aug 13, 2021 · f87fb99 · f87fb99
1 parent beb8905
commit f87fb99
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/psyneulink/core/compositions/composition.py b/psyneulink/core/compositions/composition.py
@@ -9914,7 +9914,10 @@ def _delete_contexts(self, *contexts, check_simulation_storage=False, visited=No
         super()._delete_contexts(*contexts, check_simulation_storage=check_simulation_storage, visited=visited)
 
         for c in contexts:
-            self.scheduler._delete_counts(c.execution_id)
+            try:
+                self.scheduler._delete_counts(c.execution_id)
+            except AttributeError:
+                self.scheduler._delete_counts(c)
 
     # ******************************************************************************************************************
     #                                           LLVM