Coverage for src/mlopus/mlflow/api/common/decorators.py: 93%

1import functools 

2 

3 

4def online(func): 

5 """Decorator for API methods that are unavailable in offline mode.""" 

6 

7 @functools.wraps(func) 

8 def wrapper(self, *args, **kwargs): 

9 if self.offline_mode: 

10 raise RuntimeError(f"Cannot access `{self.__class__.__name__}.{func.__name__}` in offline mode.") 

11 return func(self, *args, **kwargs) 

12 

13 return wrapper 

14 

15 

16def require_update(func): 

17 """Decorator for API methods that require updating the entity data.""" 

18 

19 @functools.wraps(func) 

20 def wrapper(self, *args, **kwargs): 

21 retval = func(self, *args, **kwargs) 

22 self.update() 

23 return retval 

24 

25 return wrapper