Coverage for encodermap/misc/function_def.py: 15%

1# -*- coding: utf-8 -*- 

2# encodermap/misc/function_def.py 

3################################################################################ 

4# Encodermap: A python library for dimensionality reduction. 

5# 

6# Copyright 2019-2022 University of Konstanz and the Authors 

7# 

8# Authors: 

9# Kevin Sawade, Tobias Lemke 

10# 

11# Encodermap is free software: you can redistribute it and/or modify 

12# it under the terms of the GNU Lesser General Public License as 

13# published by the Free Software Foundation, either version 2.1 

14# of the License, or (at your option) any later version. 

15# This package is distributed in the hope that it will be useful to other 

16# researches. IT DOES NOT COME WITH ANY WARRANTY WHATSOEVER; without even the 

17# implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. 

18# See the GNU Lesser General Public License for more details. 

19# 

20# See <http://www.gnu.org/licenses/>. 

21################################################################################ 

22 

23import tensorflow as tf 

24 

25 

26def function(debug=False): 

27 def decorator(f): 

28 def wrapper(*args, **kwargs): 

29 if debug: 

30 result = f(*args, **kwargs) 

31 else: 

32 compiled = tf.function(f) 

33 result = compiled(*args, **kwargs) 

34 return result 

35 

36 return wrapper 

37 

38 return decorator