Minor modifications to how device is selected

tests.py

@@ -31,10 +31,13 @@ class TestUtils(unittest.TestCase):
         self.assertEqual(get_gpu([True, "cpu", 0]), [0, "cpu"] if gpu_available else ["cpu", "cpu", "cpu"])
 
         # Test list input with duplicate elements
-        self.assertEqual(get_gpu([0, 0, "gpu"]), [0] if gpu_available else ["cpu", "cpu", "cpu"])
+        self.assertEqual(get_gpu([0, 0, "gpu"]), 0 if gpu_available else ["cpu", "cpu", "cpu"])
 
         # Test list input with duplicate elements of different types
-        self.assertEqual(get_gpu([True, 0, "gpu"]), [0] if gpu_available else ["cpu", "cpu", "cpu"])
+        self.assertEqual(get_gpu([True, 0, "gpu"]), 0 if gpu_available else ["cpu", "cpu", "cpu"])
+
+        # Test list input but only one element
+        self.assertEqual(get_gpu([True]), 0 if gpu_available else "cpu")
 
         # Test list input with all integers
         self.assertEqual(get_gpu(list(range(gpu_count))),

utils.py

@@ -73,7 +73,7 @@ def get_gpu(gpu: DEVICE_TYPE) -> ENCODER_DEVICE_TYPE:
                     result.append(device)
             else:
                 result.append(device)
-        return result
+        return result[0] if len(result) == 1 else result
     else:
         return _get_single_device(gpu)