Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Handle bad id in log_stats(...).

Browse Source
This commit is contained in:
Ryan Dick 2024-01-11 15:52:12 -05:00 committed by Kent Keirsey
parent aa45d21fd2
commit 296c861e7d
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
invokeai/app/services/invocation_stats/invocation_stats_default.py
View File

@ -98,8 +98,12 @@ class InvocationStatsService(InvocationStatsServiceBase):
logger.warning(f"Attempted to clear statistics for unknown graph {graph_execution_state_id}: {e}.")
def log_stats(self, graph_execution_state_id: str):
graph_stats = self._stats[graph_execution_state_id]
cache_stats = self._cache_stats[graph_execution_state_id]
try:
graph_stats = self._stats[graph_execution_state_id]
cache_stats = self._cache_stats[graph_execution_state_id]
except KeyError as e:
logger.warning(f"Attempted to log statistics for unknown graph {graph_execution_state_id}: {e}.")
return
log = graph_stats.get_pretty_log(graph_execution_state_id)