Coverage for /pythoncovmergedfiles/medio/medio/usr/local/lib/python3.8/site-packages/tensorflow/python/trackable/converter.py: 50%
16 statements
« prev ^ index » next coverage.py v7.4.0, created at 2024-01-03 07:57 +0000
« prev ^ index » next coverage.py v7.4.0, created at 2024-01-03 07:57 +0000
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"""Util for converting a Python object to a Trackable."""
18from tensorflow.python.eager.polymorphic_function import saved_model_utils
19from tensorflow.python.framework import dtypes
20from tensorflow.python.framework import tensor_util
21from tensorflow.python.ops import resource_variable_ops
22from tensorflow.python.trackable import base
23from tensorflow.python.trackable import data_structures
26def convert_to_trackable(obj, parent=None):
27 """Converts `obj` to `Trackable`."""
28 if isinstance(obj, base.Trackable):
29 return obj
30 obj = data_structures.wrap_or_unwrap(obj)
31 if (tensor_util.is_tf_type(obj) and
32 obj.dtype not in (dtypes.variant, dtypes.resource) and
33 not resource_variable_ops.is_resource_variable(obj)):
34 return saved_model_utils.TrackableConstant(obj, parent)
35 if not isinstance(obj, base.Trackable):
36 raise ValueError(f"Cannot convert {obj} to Trackable.")
37 return obj