Skip to content

Commit 546034f

Browse files
use intersection of supported_devices and device_list
1 parent 8bd5e7c commit 546034f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

demos/tests/cases.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,7 @@ def device_args(self, device_list):
5353
if len(self.device_keys) == 0:
5454
return {'CPU': []}
5555
if self.supported_devices:
56-
device_list = self.supported_devices
56+
device_list = list(set(device_list) & set(self.supported_devices))
5757
return {device: [arg for key in self.device_keys for arg in [key, device]] for device in device_list}
5858

5959
def get_models(self, case):

0 commit comments

Comments
 (0)