diff --git a/lib/plaidml_tools.py b/lib/plaidml_tools.py index 08492e1fc7..f245d6ae46 100644 --- a/lib/plaidml_tools.py +++ b/lib/plaidml_tools.py @@ -73,8 +73,8 @@ def names(self): return ["{} - {} ({})".format( device.get("vendor", "unknown"), device.get("name", "unknown"), - "supported" if device in self.supported_devices else "experimental") - for device in self.device_details] + "supported" if idx in self.supported_indices else "experimental") + for idx, device in enumerate(self.device_details)] @property def supported_indices(self):