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

fix(stats): preserve stack when raising GESStatsNotFoundError

Browse Source
This commit is contained in:
psychedelicious 2024-02-01 08:18:57 +11:00
parent b24e8dd829
commit 9f6b9d4d23
1 changed files with 4 additions and 4 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

@ -107,7 +107,7 @@ class InvocationStatsService(InvocationStatsServiceBase):
except KeyError as e:
msg = f"Attempted to clear statistics for unknown graph {graph_execution_state_id}: {e}."
logger.warning(msg)
raise GESStatsNotFoundError(msg)
raise GESStatsNotFoundError(msg) from e
def get_stats(self, graph_execution_state_id: str) -> InvocationStatsSummary:
graph_stats_summary = self._get_graph_summary(graph_execution_state_id)
@ -137,7 +137,7 @@ class InvocationStatsService(InvocationStatsServiceBase):
except KeyError as e:
msg = f"Attempted to get model cache statistics for unknown graph {graph_execution_state_id}: {e}."
logger.warning(msg)
raise GESStatsNotFoundError(msg)
raise GESStatsNotFoundError(msg) from e
return ModelCacheStatsSummary(
cache_hits=cache_stats.hits,
@ -155,7 +155,7 @@ class InvocationStatsService(InvocationStatsServiceBase):
except KeyError as e:
msg = f"Attempted to get graph statistics for unknown graph {graph_execution_state_id}: {e}."
logger.warning(msg)
raise GESStatsNotFoundError(msg)
raise GESStatsNotFoundError(msg) from e
return graph_stats.get_graph_stats_summary(graph_execution_state_id)
@ -165,6 +165,6 @@ class InvocationStatsService(InvocationStatsServiceBase):
except KeyError as e:
msg = f"Attempted to get node statistics for unknown graph {graph_execution_state_id}: {e}."
logger.warning(msg)
raise GESStatsNotFoundError(msg)
raise GESStatsNotFoundError(msg) from e
return graph_stats.get_node_stats_summaries()