mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

fix sampler call in affine transform test

This commit is contained in:
mohammad 2021-02-26 15:35:06 -05:00
parent f1c7aeaa6b
commit 6ffaf6bf00
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/ut/cpp/dataset/c_api_affine_test.cc
View File

@ -31,7 +31,7 @@ TEST_F(MindDataTestPipeline, TestAffineAPI) {
// Create an ImageFolder Dataset
std::string folder_path = datasets_root_path_ + "/testPK/data/";
std::shared_ptr<Dataset> ds = ImageFolder(folder_path, true, RandomSampler(false, 5));
std::shared_ptr<Dataset> ds = ImageFolder(folder_path, true, std::make_shared<RandomSampler>(false, 5));
// Create a Repeat operation on ds
int32_t repeat_num = 3;
@ -74,7 +74,7 @@ TEST_F(MindDataTestPipeline, TestAffineAPIFail) {
// Create an ImageFolder Dataset
std::string folder_path = datasets_root_path_ + "/testPK/data/";
std::shared_ptr<Dataset> ds = ImageFolder(folder_path, true, RandomSampler(false, 5));
std::shared_ptr<Dataset> ds = ImageFolder(folder_path, true, std::make_shared<RandomSampler>(false, 5));
// Create a Repeat operation on ds
int32_t repeat_num = 3;