Coverage for /pythoncovmergedfiles/medio/medio/usr/local/lib/python3.8/site-packages/tensorflow/python/keras/utils/tf_contextlib.py: 100%

1# Copyright 2017 The TensorFlow Authors. All Rights Reserved. 

2# 

3# Licensed under the Apache License, Version 2.0 (the "License"); 

4# you may not use this file except in compliance with the License. 

5# You may obtain a copy of the License at 

6# 

7# http://www.apache.org/licenses/LICENSE-2.0 

8# 

9# Unless required by applicable law or agreed to in writing, software 

10# distributed under the License is distributed on an "AS IS" BASIS, 

11# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 

12# See the License for the specific language governing permissions and 

13# limitations under the License. 

14# ============================================================================== 

15"""TFDecorator-aware replacements for the contextlib module.""" 

16 

17import contextlib as _contextlib 

18 

19from tensorflow.python.util import tf_decorator 

20 

21 

22def contextmanager(target): 

23 """A tf_decorator-aware wrapper for `contextlib.contextmanager`. 

24 

25 Usage is identical to `contextlib.contextmanager`. 

26 

27 Args: 

28 target: A callable to be wrapped in a contextmanager. 

29 Returns: 

30 A callable that can be used inside of a `with` statement. 

31 """ 

32 context_manager = _contextlib.contextmanager(target) 

33 return tf_decorator.make_decorator(target, context_manager, 'contextmanager')