Browse Source

!12676 Fix sampler signature in affine transform test

From: @mhmotallebi
Reviewed-by: @nsyca,@robingrosman
Signed-off-by: @robingrosman
tags/v1.2.0-rc1
mindspore-ci-bot Gitee 4 years ago
parent
commit
70e7877155
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