Accomodating PR comments.

microsoft · Nov 27, 2018 · 0bb5cb7 · 0bb5cb7
1 parent e7e801b
commit 0bb5cb7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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));