Browse Source

fix sampler call in affine transform test

tags/v1.2.0-rc1
mohammad 5 years ago
parent
commit
6ffaf6bf00
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      tests/ut/cpp/dataset/c_api_affine_test.cc

+ 2
- 2
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;


Loading…
Cancel
Save