fix ruff error

Author: liurt1218

traincheck/instrumentor/source_file.py

@@ -618,8 +618,8 @@ def at_attr(name: str) -> bool:
     new_src = "".join(new_lines)
 
     def _find_annotate_import_idx(lines):
-        for idx, l in enumerate(lines):
-            if re.match(r"^\s*from\s+traincheck\s+import\s+annotate_stage\s*$", l):
+        for idx, line in enumerate(lines):
+            if re.match(r"^\s*from\s+traincheck\s+import\s+annotate_stage\s*$", line):
                 return idx
         return -1
 

traincheck/trace/trace_pandas.py

@@ -183,6 +183,11 @@ def _rm_incomplete_trailing_func_calls(self):
             self.events.groupby("func_call_id").size().reset_index(name="count")
         )
 
+        multiple_func_call_ids = func_call_groups[func_call_groups["count"] >2][
+            "func_call_id"
+        ]
+        assert len(multiple_func_call_ids)==0, "more than 2 events for one func call id"
+        
         incomplete_func_call_ids = func_call_groups[func_call_groups["count"] == 1][
             "func_call_id"
         ]