diff --git a/test/utils.py b/test/utils.py index 79562256..2d14eb40 100644 --- a/test/utils.py +++ b/test/utils.py @@ -14,7 +14,7 @@ def get_available_devices() -> List[device]: """Return CPU and, if present, GPU device. - AdaptiveAvgPool2d, Conv2d, Flatten, + Returns: devices: Available devices for ``torch``. """