Fix not returning values from safe_write functions

2024-09-22 14:20:34 +01:00 · 2024-09-22 14:20:34 +01:00 · 2ad7ed2ff2
commit 2ad7ed2ff2
parent 93590d14b7
1 changed files with 2 additions and 1 deletions
--- a/matchy/state.py
+++ b/matchy/state.py
@ -232,11 +232,12 @@ class _State():
        @wraps(func)
        def inner(self, *args, **kwargs):
            tmp = _State(self._dict, self._file)
-            func(tmp, *args, **kwargs)
+            ret = func(tmp, *args, **kwargs)
            _SCHEMA.validate(tmp._dict)
            if tmp._file:
                _save(tmp._file, tmp._dict)
            self._dict = tmp._dict
+            return ret

        return inner