Skip to content

Commit 1374fd2

Browse files
committed
Lint check
Signed-off-by: Abhishek <maurya.abhishek@ibm.com>
1 parent 8f51d3b commit 1374fd2

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

tests/models/test_decoders.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -69,7 +69,9 @@
6969
USE_MICRO_MODELS = os.environ.get("FMS_TEST_SHAPES_USE_MICRO_MODELS", "1") == "1"
7070
USE_DISTRIBUTED = os.environ.get("FMS_TEST_SHAPES_DISTRIBUTED", "0") == "1"
7171
TIMING = os.environ.get("TIMING", "")
72-
CUMULATIVE_TEST_TOKENS_PER_SEQUENCE = os.environ.get("FMS_TEST_SHAPES_CUMULATIVE_TEST_TOKENS_PER_SEQUENCE", "1024")
72+
CUMULATIVE_TEST_TOKENS_PER_SEQUENCE = os.environ.get(
73+
"FMS_TEST_SHAPES_CUMULATIVE_TEST_TOKENS_PER_SEQUENCE", "1024"
74+
)
7375
ATTN_TYPE = os.environ.get("FMS_TEST_SHAPES_ATTN_TYPE", "sdpa")
7476
attention_map = {
7577
"sdpa": "sdpa_causal",

0 commit comments

Comments
 (0)