Clarify comments for RESTORMER_CONFIGS in test_restormer.py,

I, Cano-Muniz, Santiago <[email protected]>, hereby add my Signed-off-by to this commit: 3c2dbc6 Signed-off-by: Cano-Muniz, Santiago <[email protected]>
Project-MONAI · Mar 8, 2025 · 4573ec9 · 4573ec9
1 parent f17e06e
commit 4573ec9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/networks/nets/test_restormer.py b/tests/networks/nets/test_restormer.py
@@ -40,10 +40,10 @@
 ]
 
 RESTORMER_CONFIGS = [
-    # 2-level architecture
+    # 2-level architecture test
     {"num_blocks": [1, 1], "heads": [1, 1]},
     {"num_blocks": [2, 1], "heads": [2, 1]},
-    # 3-level architecture
+    # 3-level architecture test
     {"num_blocks": [1, 1, 1], "heads": [1, 1, 1]},
     {"num_blocks": [2, 1, 1], "heads": [2, 1, 1]},
 ]