Fix nightly build

PiperOrigin-RevId: 236139067
This commit is contained in:
Rachel Lim 2019-02-28 09:42:28 -08:00 committed by TensorFlower Gardener
parent 42a1de008f
commit fb31b9fde4
2 changed files with 6 additions and 4 deletions

View File

@ -345,8 +345,9 @@ TEST_P(ParameterizedDatasetTest, Roundtrip) {
} }
} }
INSTANTIATE_TEST_SUITE_P(ConcatenateDatasetOpTest, ParameterizedDatasetTest, INSTANTIATE_TEST_SUITE_P(
::testing::ValuesIn({TestCase1(), TestCase2()})); ConcatenateDatasetOpTest, ParameterizedDatasetTest,
::testing::ValuesIn(std::vector<TestParam>({TestCase1(), TestCase2()})));
} // namespace } // namespace
} // namespace data } // namespace data
} // namespace tensorflow } // namespace tensorflow

View File

@ -344,8 +344,9 @@ TEST_P(ParameterizedDatasetTest, Roundtrip) {
} }
} }
INSTANTIATE_TEST_SUITE_P(ZipDatasetOpTest, ParameterizedDatasetTest, INSTANTIATE_TEST_SUITE_P(
::testing::ValuesIn({TestCase1(), TestCase2()})); ZipDatasetOpTest, ParameterizedDatasetTest,
::testing::ValuesIn(std::vector<TestParam>({TestCase1(), TestCase2()})));
} // namespace } // namespace
} // namespace data } // namespace data