[Serve] Address incremental memory leak due to _PyObjScanner (#31317)

ray-project · Dec 23, 2022 · 01b19ba · 01b19ba
1 parent 1e42e6c
commit 01b19ba
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/python/ray/dag/dag_node.py b/python/ray/dag/dag_node.py
@@ -175,6 +175,7 @@ def _get_all_child_nodes(self) -> List["DAGNode"]:
         ):
             if n not in children:
                 children.append(n)
+        scanner.clear()
         return children
 
     def _apply_and_replace_all_child_nodes(
@@ -210,6 +211,7 @@ def _apply_and_replace_all_child_nodes(
         new_args, new_kwargs, new_other_args_to_resolve = scanner.replace_nodes(
             replace_table
         )
+        scanner.clear()
 
         # Return updated copy of self.
         return self._copy(
@@ -288,6 +290,7 @@ def apply_functional(
                 replace_table[node] = apply_fn(node)
 
         replaced_inputs = scanner.replace_nodes(replace_table)
+        scanner.clear()
 
         return replaced_inputs