m53297601
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

!12676 Fix sampler signature in affine transform test

Browse Source 
From: @mhmotallebi
Reviewed-by: @nsyca,@robingrosman
Signed-off-by: @robingrosman
pull/12676/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
parent d8433e9f5c 6ffaf6bf00
commit 70e7877155
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File

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

@ -31,7 +31,7 @@ TEST_F(MindDataTestPipeline, TestAffineAPI) {
// Create an ImageFolder Dataset // Create an ImageFolder Dataset
std::string folder_path = datasets_root_path_ + "/testPK/data/"; 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 // Create a Repeat operation on ds
int32_t repeat_num = 3; int32_t repeat_num = 3;
@ -74,7 +74,7 @@ TEST_F(MindDataTestPipeline, TestAffineAPIFail) {
// Create an ImageFolder Dataset // Create an ImageFolder Dataset
std::string folder_path = datasets_root_path_ + "/testPK/data/"; 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 // Create a Repeat operation on ds
int32_t repeat_num = 3; int32_t repeat_num = 3;

Write Preview
Loading…
Cancel
Save