From eb36c213032c92f64bab2d7ac3a9841f48724eb9 Mon Sep 17 00:00:00 2001 From: frostedoyster Date: Sun, 28 Jul 2024 15:37:16 +0200 Subject: [PATCH] Small docstring change --- src/metatrain/utils/devices.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/metatrain/utils/devices.py b/src/metatrain/utils/devices.py index 937f3540..3927260b 100644 --- a/src/metatrain/utils/devices.py +++ b/src/metatrain/utils/devices.py @@ -22,8 +22,8 @@ def pick_devices( :param architecture_devices: Devices supported by the architecture. The list should be sorted by the preference of the architecture while the most prefferred device should be first and the least one last. - :param desired_device: desired device by the user i.e. ``"cpu"``, "``cuda``" or - ``"multi-gpu"`` + :param desired_device: desired device by the user. For example, ``"cpu"``, "``cuda``", + ``"multi-gpu"``, etc. """ available_devices = ["cpu"]