fix: do not supress reraise in audit wrapper

z3z1ma · Jul 12, 2024 · abf0e82 · abf0e82
1 parent 41c23f7
commit abf0e82
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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))