o i e @sdZddlmZddlmZddlmZddlmZddl m Z ddl m Z ddl m Z ddl mZdd lmZd d Zd d ZddZdS)zUtilities for SavedModel fingerprinting. This module contains utility classes and functions for working with the SavedModel fingerprint. )logging)flags)fingerprint_pb2)file_io)fingerprinting) constants)metrics)compatc CstjrKtt|ttj }t d|zt |}Wnt jy3}zt|dd}~wwt||tj|dtj|t|ddSdS)a!Write fingerprint protobuf, if requested. Writes a `tf.saved_model.experimental.Fingerprint` object to a `fingerprint.pb` file in the `export_dir` using the `saved_model.pb` file contained in `export_dir`. Args: export_dir: The directory in which to write the fingerprint. zWriting fingerprint to %sN fingerprint)path singleprint)rconfigsaved_model_fingerprintingvaluerjoinr as_strrFINGERPRINT_FILENAMErinfofingerprinting_pywrapCreateFingerprintDefFingerprintException ValueErroratomic_write_string_to_filerSetWriteFingerprintSetWritePathAndSingleprintsingleprint_from_saved_model) export_dirfingerprint_pathfingerprint_serializeder!a/var/www/myenv/lib/python3.10/site-packages/tensorflow/python/saved_model/fingerprinting_utils.pywrite_fingerprint!s.      r#c Cs2zt|WStjy}zt|dd}~ww)a2Returns the singleprint of a SavedModel in `export_dir`. Args: export_dir: The directory that contains the SavedModel. Returns: A string containing the singleprint of the SavedModel in `export_dir`. Raises: ValueError: If a valid singleprint cannot be constructed from a SavedModel. N)rSingleprintFromSMrr)rr r!r!r"r?s   rcCs2t|tjs tdtj|j|j|j|j |j dS)Nz$Supplied value is not a Fingerprint.)saved_model_checksumgraph_def_program_hashsignature_def_hashsaved_object_graph_hashcheckpoint_hash) isinstancer Fingerprint TypeErrorrFingerprintDefr%r&r'r(r)r r!r!r"to_protoQs r.N)__doc__abslrtensorflow.core.configrtensorflow.core.protobufrtensorflow.python.lib.iortensorflow.python.saved_modelr0tensorflow.python.saved_model.pywrap_saved_modelrrrtensorflow.python.utilr r#rr.r!r!r!r"s