Clarify comments for RESTORMER_CONFIGS in test_restormer.py,

phisanti · phisanti · commit 4573ec924749 · 2025-03-08T20:57:36.000+01:00
I, Cano-Muniz, Santiago <tisalon@outlook.com>, hereby add my Signed-off-by to this commit: 3c2dbc6 Signed-off-by: Cano-Muniz, Santiago <tisalon@outlook.com>
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]},
 ]

Original file line number	Diff line number	Diff line change
`@@ -40,10 +40,10 @@`
`40`	`40`	`]`
`41`	`41`
`42`	`42`	`RESTORMER_CONFIGS = [`
`43`		`- # 2-level architecture`
	`43`	`+ # 2-level architecture test`
`44`	`44`	`{"num_blocks": [1, 1], "heads": [1, 1]},`
`45`	`45`	`{"num_blocks": [2, 1], "heads": [2, 1]},`
`46`		`- # 3-level architecture`
	`46`	`+ # 3-level architecture test`
`47`	`47`	`{"num_blocks": [1, 1, 1], "heads": [1, 1, 1]},`
`48`	`48`	`{"num_blocks": [2, 1, 1], "heads": [2, 1, 1]},`
`49`	`49`	`]`