make format

matsengrp · Nov 13, 2024 · 039d7be · 039d7be
1 parent 84f17cd
commit 039d7be
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/netam/common.py b/netam/common.py
@@ -204,9 +204,9 @@ def pick_device(gpu_preference=None):
     If gpu_preference is a string, use the device with that name. This is considered a
     strong preference from a user who knows what they are doing.
 
-    If gpu_preference is an integer, this is a weak preference for a numbered
-    GPU.  If CUDA is available, use the least used GPU, and if all are idle use
-    the gpu_index modulo the number of GPUs. If gpu_index is None, then use a random GPU.
+    If gpu_preference is an integer, this is a weak preference for a numbered GPU.  If
+    CUDA is available, use the least used GPU, and if all are idle use the gpu_index
+    modulo the number of GPUs. If gpu_index is None, then use a random GPU.
     """
 
     # Strong preference for a specific device.