github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-04-23 03:00:31 -04: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
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)