Revert to full proto

2020-12-30 18:47:30 -08:00 · 2020-12-30 18:47:30 -08:00 · dd5f07f812
commit dd5f07f812
parent 251e710762
1 changed files with 16 additions and 0 deletions
--- a/tensorflow/python/kernel_tests/boosted_trees/resource_ops_test.py
+++ b/tensorflow/python/kernel_tests/boosted_trees/resource_ops_test.py
@ -129,6 +129,22 @@ class ResourceOpsTest(test_util.TensorFlowTestCase):
            }
          }
        }
+        tree_weights: 0.15
+        tree_weights: 1.0
+        tree_metadata {
+          num_layers_grown: 2
+          is_finalized: true
+        }
+        tree_metadata {
+          num_layers_grown: 1
+          is_finalized: false
+        }
+        growing_metadata {
+          num_trees_attempted: 2
+          num_layers_attempted: 6
+          last_layer_node_start: 16
+          last_layer_node_end: 19
+        }
      """, ensemble_proto)
      ensemble = boosted_trees_ops.TreeEnsemble(
          'ensemble',