From 569f6ed488afe1aeaa1aaccf8b9da10462e2e132 Mon Sep 17 00:00:00 2001 From: runame Date: Fri, 12 Jan 2024 18:19:21 +0100 Subject: [PATCH] Fix docstring --- test/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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``. """