KeyArgo
/
smolagents
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Align step numbers in monitoring logging (#667)

This commit is contained in:
spemed 2025-02-17 16:22:25 +08:00 committed by GitHub
parent 345f1c35ca
commit 1a272f6a9e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/smolagents/monitoring.py
View File

@ -59,7 +59,7 @@ class Monitor:
""" """
step_duration = step_log.duration step_duration = step_log.duration
self.step_durations.append(step_duration) self.step_durations.append(step_duration)
console_outputs = f"[Step {len(self.step_durations) - 1}: Duration {step_duration:.2f} seconds" console_outputs = f"[Step {len(self.step_durations)}: Duration {step_duration:.2f} seconds"
if getattr(self.tracked_model, "last_input_token_count", None) is not None: if getattr(self.tracked_model, "last_input_token_count", None) is not None:
self.total_input_token_count += self.tracked_model.last_input_token_count self.total_input_token_count += self.tracked_model.last_input_token_count