Deprecated tf.Session removed in generate_examples_lib.py

This commit is contained in:
Siju Samuel 2019-07-10 14:43:17 +05:30
parent 48959a13a1
commit 07ac3694dd

View File

@ -625,7 +625,7 @@ def make_zip_of_tests(options,
report["tf_log"] += traceback.format_exc() report["tf_log"] += traceback.format_exc()
return None, report return None, report
sess = tf.Session() sess = tf.compat.v1.Session()
try: try:
baseline_inputs, baseline_outputs = (make_test_inputs( baseline_inputs, baseline_outputs = (make_test_inputs(
param_dict_real, sess, inputs, outputs)) param_dict_real, sess, inputs, outputs))