提交 c0d18d25 编写于 作者: X xiefangqi

fix randomdata ut problem

上级 e2a63790
......@@ -100,11 +100,8 @@ TEST_F(MindDataTestPipeline, TestRandomDatasetBasic2) {
// Check if RandomDataOp read correct columns
uint64_t i = 0;
while (row.size() != 0) {
auto image = row["image"];
auto label = row["label"];
MS_LOG(INFO) << "Tensor image shape: " << image->shape();
MS_LOG(INFO) << "Tensor label shape: " << label->shape();
// If no schema specified, RandomData will generate random columns
// So we don't check columns here
iter->GetNextRow(&row);
i++;
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册