Skip to content

Commit

Permalink
Small docstring change
Browse files Browse the repository at this point in the history
  • Loading branch information
frostedoyster committed Jul 28, 2024
1 parent ab3d1c3 commit eb36c21
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions src/metatrain/utils/devices.py
Original file line number Diff line number Diff line change
Expand Up @@ -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"]
Expand Down

0 comments on commit eb36c21

Please sign in to comment.