Coverage for /pythoncovmergedfiles/medio/medio/usr/local/lib/python3.8/site-packages/tensorflow/python/util/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.""" 

16import contextlib as _contextlib 

17 

18from tensorflow.python.util import tf_decorator 

19 

20 

21def contextmanager(target): 

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

23 

24 Usage is identical to `contextlib.contextmanager`. 

25 

26 Args: 

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

28 Returns: 

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

30 """ 

31 context_manager = _contextlib.contextmanager(target) 

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