From c0b546a96eaaaaef2e9ab1bf32b1abe9092d33e1 Mon Sep 17 00:00:00 2001
From: "huizheng.hz" <huizheng.hz@alibaba-inc.com>
Date: Tue, 18 Oct 2022 14:34:26 +0800
Subject: [PATCH] [to #42322933]add subset_name when loading dataset (NAFNet
 image denoising)         Link:
 https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/10427797

---
 tests/trainers/test_image_denoise_trainer.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tests/trainers/test_image_denoise_trainer.py b/tests/trainers/test_image_denoise_trainer.py
index 0bcb8930..68ddf616 100644
--- a/tests/trainers/test_image_denoise_trainer.py
+++ b/tests/trainers/test_image_denoise_trainer.py
@@ -33,11 +33,13 @@ class ImageDenoiseTrainerTest(unittest.TestCase):
         dataset_train = MsDataset.load(
             'SIDD',
             namespace='huizheng',
+            subset_name='default',
             split='validation',
             download_mode=DownloadMode.REUSE_DATASET_IF_EXISTS)._hf_ds
         dataset_val = MsDataset.load(
             'SIDD',
             namespace='huizheng',
+            subset_name='default',
             split='test',
             download_mode=DownloadMode.REUSE_DATASET_IF_EXISTS)._hf_ds
         self.dataset_train = SiddImageDenoisingDataset(