fix sampler call in affine transform test

4 years ago · 6ffaf6bf00
parent f1c7aeaa6b
commit 6ffaf6bf00
1 changed files with 2 additions and 2 deletions
--- a/tests/ut/cpp/dataset/c_api_affine_test.cc
+++ b/tests/ut/cpp/dataset/c_api_affine_test.cc
@ -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;