@@ -209,10 +209,7 @@ def test_check_grad(self):
 
         place = core.CUDAPlace(0)
         self.check_grad_with_place(
-            place,
-            ['X', 'Scale', 'Bias'],
-            'Y',
-            check_pir=True,
+            place, {'X', 'Scale', 'Bias'}, 'Y', check_pir=True
         )
 
     def init_test_case(self):
@@ -279,10 +276,7 @@ def test_check_grad(self):
 
         place = core.CUDAPlace(0)
         self.check_grad_with_place(
-            place,
-            ['X', 'Scale', 'Bias'],
-            'Y',
-            check_pir=True,
+            place, {'X', 'Scale', 'Bias'}, 'Y', check_pir=True
         )
 
     def init_test_case(self):