more memoization bugfixes

ipyflow · Nov 29, 2023 · ac78cf8 · ac78cf8
1 parent 4c5ca0d
commit ac78cf8
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/core/ipyflow/data_model/cell.py b/core/ipyflow/data_model/cell.py
@@ -505,6 +505,7 @@ def to_ast(self, override: Optional[ast.Module] = None) -> ast.Module:
             return self._cached_ast
         if (
             self._cached_ast is None
+            or self.last_ast_content is None
             or len(self.last_ast_content) != len(self.current_content)
             or self.last_ast_content != self.current_content
         ):

diff --git a/core/ipyflow/shell/interactiveshell.py b/core/ipyflow/shell/interactiveshell.py
@@ -517,8 +517,8 @@ def before_run_cell(
                 if sym.obj is not value:
                     self.user_ns[sym.name] = value
                     sym.update_obj_ref(value)
-                    new_updated_ts = Timestamp(self.cell_counter(), out_ts.stmt_num)
-                    sym.refresh(timestamp=new_updated_ts)
+                new_updated_ts = Timestamp(self.cell_counter(), out_ts.stmt_num)
+                sym.refresh(timestamp=new_updated_ts)
             return f"Out.get({identical_result_ctr})"
 
         # Stage 1: Precheck.