o ?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)pathZ singleprint)rconfigZsaved_model_fingerprintingvaluerjoinr Zas_strrZFINGERPRINT_FILENAMErinfofingerprinting_pywrapZCreateFingerprintDefFingerprintException ValueErrorZatomic_write_string_to_filerZSetWriteFingerprintZSetWritePathAndSingleprintsingleprint_from_saved_model) export_dirZfingerprint_pathZfingerprint_serializeders/home/www/facesmatcher.com/pyenv/lib/python3.10/site-packages/tensorflow/python/saved_model/fingerprinting_utils.pywrite_fingerprint!s.      rc 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)rZSingleprintFromSMrr)rrrrrr?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) isinstancerZ Fingerprint TypeErrorrZFingerprintDefrrrrrr rrrto_protoQs r!N)__doc__ZabslrZtensorflow.core.configrZtensorflow.core.protobufrZtensorflow.python.lib.iorZtensorflow.python.saved_modelrZ0tensorflow.python.saved_model.pywrap_saved_modelrrrZtensorflow.python.utilr rrr!rrrrs