Coverage for /pythoncovmergedfiles/medio/medio/usr/local/lib/python3.8/site-packages/tensorflow/core/function/trace_type/util.py: 62%

1# Copyright 2022 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"""Utilities for the trace_type module.""" 

16 

17import numpy as np 

18 

19 

20# TODO(b/225045380): Depend on the abstracted `leaf` lib from 'nest'. 

21def is_namedtuple(obj): 

22 return hasattr(obj, "_fields") and all( 

23 isinstance(field, str) for field in obj._fields) 

24 

25 

26# TODO(b/225045380): Depend on the abstracted `leaf` lib from 'nest'. 

27def is_attrs(obj): 

28 return hasattr(type(obj), "__attrs_attrs__") 

29 

30 

31# TODO(b/225045380): Depend on the abstracted `leaf` lib from 'nest'. 

32def is_np_ndarray(value): 

33 return hasattr(value, "__array__") and not ( 

34 # For legacy reasons we do not automatically promote Numpy strings. 

35 isinstance(value, np.str_) 

36 # NumPy dtypes have __array__ as unbound methods. 

37 or isinstance(value, type))