From abf0e82de7d64cde7a76491c5c2bc46b71efe719 Mon Sep 17 00:00:00 2001 From: z3z1ma Date: Fri, 12 Jul 2024 09:54:01 -0700 Subject: [PATCH] fix: do not supress reraise in audit wrapper --- src/cdf/core/feature_flag/file.py | 2 +- src/cdf/core/state.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/cdf/core/feature_flag/file.py b/src/cdf/core/feature_flag/file.py index a1503f0..cd681a4 100644 --- a/src/cdf/core/feature_flag/file.py +++ b/src/cdf/core/feature_flag/file.py @@ -69,7 +69,7 @@ def _flags(self) -> t.Dict[str, FlagAdapterResponse]: """Get the feature flags.""" if self.__flags is None: self.__flags = self._read() - return self.__flags + return t.cast(t.Dict[str, FlagAdapterResponse], self.__flags) def get(self, feature_name: str) -> FlagAdapterResponse: """Get a feature flag. diff --git a/src/cdf/core/state.py b/src/cdf/core/state.py index e049e2a..ed5e357 100644 --- a/src/cdf/core/state.py +++ b/src/cdf/core/state.py @@ -204,7 +204,7 @@ def wrapper(*args: P.args, **kwargs: P.kwargs) -> T: self.audit_table, pd.DataFrame([audit_event]), ) - raise e + raise e audit_event["elapsed"] = time.perf_counter() - start audit_event["success"] = not isinstance(rv, M.Err) audit_event["properties"].update(output_props(rv))