diff --git a/tensorflow/python/kernel_tests/boosted_trees/resource_ops_test.py b/tensorflow/python/kernel_tests/boosted_trees/resource_ops_test.py index 81078bd5b83..4f69424386d 100644 --- a/tensorflow/python/kernel_tests/boosted_trees/resource_ops_test.py +++ b/tensorflow/python/kernel_tests/boosted_trees/resource_ops_test.py @@ -39,127 +39,10 @@ class ResourceOpsTest(test_util.TensorFlowTestCase): ensemble = boosted_trees_ops.TreeEnsemble('ensemble') resources.initialize_resources(resources.shared_resources()).run() _p("testCreate 2", file=sys.stderr) - - @test_util.run_deprecated_v1 - def testCreateWithProto(self): - _p("testCreateWithProto 1", file=sys.stderr) with self.cached_session(): - _p("testCreateWithProto 2", file=sys.stderr) - ensemble_proto = boosted_trees_pb2.TreeEnsemble() - _p("testCreateWithProto 3", file=sys.stderr) - text_format.Merge( - """ - trees { - nodes { - bucketized_split { - feature_id: 4 - left_id: 1 - right_id: 2 - } - metadata { - gain: 7.62 - } - } - nodes { - bucketized_split { - threshold: 21 - left_id: 3 - right_id: 4 - } - metadata { - gain: 1.4 - original_leaf { - scalar: 7.14 - } - } - } - nodes { - bucketized_split { - feature_id: 1 - threshold: 7 - left_id: 5 - right_id: 6 - } - metadata { - gain: 2.7 - original_leaf { - scalar: -4.375 - } - } - } - nodes { - leaf { - scalar: 6.54 - } - } - nodes { - leaf { - scalar: 7.305 - } - } - nodes { - leaf { - scalar: -4.525 - } - } - nodes { - leaf { - scalar: -4.145 - } - } - } - trees { - nodes { - bucketized_split { - feature_id: 75 - threshold: 21 - left_id: 1 - right_id: 2 - } - metadata { - gain: -1.4 - } - } - nodes { - leaf { - scalar: -0.6 - } - } - nodes { - leaf { - scalar: 0.165 - } - } - } - 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) - _p("testCreateWithProto 4", file=sys.stderr) - ensemble = boosted_trees_ops.TreeEnsemble( - 'ensemble', - stamp_token=7, - serialized_proto=ensemble_proto.SerializeToString()) - _p("testCreateWithProto 5", file=sys.stderr) - r = resources.shared_resources() - _p("testCreateWithProto 6", file=sys.stderr) - r=resources.initialize_resources(r) - _p("testCreateWithProto 7", file=sys.stderr) - #r.run() - #_p("testCreateWithProto 8", file=sys.stderr) + ensemble = boosted_trees_ops.TreeEnsemble('ensemble') + resources.initialize_resources(resources.shared_resources()).run() + _p("testCreate 3", file=sys.stderr) if __name__ == '__main__':