Skip to content

Commit

Permalink
Doc change
Browse files Browse the repository at this point in the history
  • Loading branch information
kbolashev committed May 9, 2024
1 parent 1cdcbf5 commit 26d7b78
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion dagshub/mlflow/patch.py
Original file line number Diff line number Diff line change
Expand Up @@ -50,7 +50,8 @@ class MlflowMonkeyPatch:
def __init__(self, funcs_to_patch: List[str], guaranteed_raises: List[str]):
self.funcs_to_patch = set(funcs_to_patch)
# Functions that guarantee to raise an exception if they are in the stack
# Example: all mlflow.<framework>.log_model functions underneath use log_artifact, but we might want to raise them
# Example: all mlflow.<framework>.log_model functions underneath use log_artifact,
# but we might want to raise exceptions, so logging models doesn't get swallowed
self.functions_with_guaranteed_raises = set(guaranteed_raises)

self.original_func_lookup: Dict[str, Callable] = {}
Expand Down

0 comments on commit 26d7b78

Please sign in to comment.