fixed formatting; fixed issue when attn_name not specified

Signed-off-by: Joshua Rosenkranz <jmrosenk@us.ibm.com>

Author: JRosenkranz

aiu_fms_testing_utils/testing/validation.py

@@ -3,8 +3,8 @@
 
 import torch
 from aiu_fms_testing_utils.utils.aiu_setup import dprint
-import os
 from aiu_fms_testing_utils._version import version_tuple
+import os
 
 
 class LogitsExtractorHook(
@@ -261,7 +261,7 @@ def extract_validation_information(
     **extra_kwargs,
 ):
     attention_specific_kwargs = {}
-    if "paged" in extra_kwargs["attn_name"]:
+    if "paged" in extra_kwargs.get("attn_name", "sdpa"):
         from aiu_fms_testing_utils.utils.paged import generate
     else:
         # TODO: Add a unified generation dependent on attn_type

tests/testing/test_validation.py

@@ -11,8 +11,8 @@
 from aiu_fms_testing_utils._version import version_tuple
 from fms.models import get_model
 from fms.utils.generation import pad_input_ids
-import torch
 from pathlib import Path
+import torch
 
 
 @pytest.mark.parametrize(