update

sot/opcode_translator/transform.py

@@ -81,8 +81,9 @@ def eval_frame_callback(frame, **kwargs) -> CustomCode:
                 used_code = custom_code.code
 
         # just check those codes which need open eval_frame
-        if custom_code.disable_eval_frame is False and CodeStatus().check_code(
-            used_code
+        if (
+            custom_code.disable_eval_frame is False
+            and CodeStatus().is_code_without_graph(used_code)
         ):
             log(
                 3,

sot/utils/code_status.py

@@ -42,7 +42,7 @@ def clear(self):
         self.code_map.clear()
         self.setup_code_map()
 
-    def check_code(self, code):
+    def is_code_without_graph(self, code):
         if code not in self.code_map:
             info = CodeInfo()
             self.code_map[code] = info
@@ -51,17 +51,16 @@ def check_code(self, code):
 
         if info.state == CodeState.WITHOUT_GRAPH:
             return True
-        elif info.state == CodeState.UNKNOW:
-            self.visit(code)
+        if info.state == CodeState.UNKNOW:
+            info.counter += 1
+            if info.counter >= 10:
+                log(
+                    3,
+                    f"[CodeStatus] Switch state to WITHOUT_GRAPH for {code}\n",
+                )
+                info.state = CodeState.WITHOUT_GRAPH
         return False
 
-    def visit(self, code):
-        info = self.code_map[code]
-        info.counter += 1
-        if info.counter >= 10:
-            log(3, f"[CodeStatus] Switch state to WITHOUT_GRAPH for {code}\n")
-            info.state = CodeState.WITHOUT_GRAPH
-
     def trace_back_frames(self):
         frame = inspect.currentframe()
         while frame.f_back is not None: