From 0bb5cb7a3075905f6d9894151b76f4dffdf295c4 Mon Sep 17 00:00:00 2001
From: Du Li <duli1@microsoft.com>
Date: Mon, 26 Nov 2018 21:02:18 -0800
Subject: [PATCH] Accomodating PR comments.

---
 onnxruntime/test/contrib_ops/expand_dims_test.cc | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/onnxruntime/test/contrib_ops/expand_dims_test.cc b/onnxruntime/test/contrib_ops/expand_dims_test.cc
index 335a4d3e8f62d..abf1da0941f8f 100644
--- a/onnxruntime/test/contrib_ops/expand_dims_test.cc
+++ b/onnxruntime/test/contrib_ops/expand_dims_test.cc
@@ -9,7 +9,7 @@ namespace test {
 
 TEST(ContribOpTest, ExpandDims_0) {
   OpTester test("ExpandDims", 1, onnxruntime::kMSDomain);
-  test.AddShapeToTensorData(true);  // TODO: re-enable shape inference test
+  test.AddShapeToTensorData(true);
   test.AddInput<float>("X", {2, 3}, std::vector<float>(6, 1.0f));
   test.AddInput<int32_t>("axis", {}, {-1});
   test.AddOutput<float>("Y", {2, 3, 1}, std::vector<float>(6, 1.0f));
@@ -18,7 +18,7 @@ TEST(ContribOpTest, ExpandDims_0) {
 
 TEST(ContribOpTest, ExpandDims_1) {
   OpTester test("ExpandDims", 1, onnxruntime::kMSDomain);
-  test.AddShapeToTensorData(true);  // TODO: re-enable shape inference test
+  test.AddShapeToTensorData(true);
   test.AddInput<float>("X", {2, 3}, std::vector<float>(6, 1.0f));
   test.AddInput<int32_t>("axis", {}, {1});
   test.AddOutput<float>("Y", {2, 1, 3}, std::vector<float>(6, 1.0f));