Spaces:
Sleeping
Sleeping
test(core): update HRM test case to correct structure
Browse files
tests/test_model_support.py
CHANGED
|
@@ -39,12 +39,12 @@ def test_model_support_rejects_non_eager_attention() -> None:
|
|
| 39 |
def test_model_support_accepts_hrm_layout() -> None:
|
| 40 |
layer = SimpleNamespace(attention=object())
|
| 41 |
model = SimpleNamespace(
|
| 42 |
-
model=SimpleNamespace(h_stack=
|
| 43 |
config=SimpleNamespace(_attn_implementation="eager"),
|
| 44 |
)
|
| 45 |
|
| 46 |
support = describe_model_support(model)
|
| 47 |
|
| 48 |
assert support.supports_attribution is True
|
| 49 |
-
assert support.layer_path == "model.h_stack
|
| 50 |
assert support.attention_attr == "attention"
|
|
|
|
| 39 |
def test_model_support_accepts_hrm_layout() -> None:
|
| 40 |
layer = SimpleNamespace(attention=object())
|
| 41 |
model = SimpleNamespace(
|
| 42 |
+
model=SimpleNamespace(h_stack=[layer, layer]),
|
| 43 |
config=SimpleNamespace(_attn_implementation="eager"),
|
| 44 |
)
|
| 45 |
|
| 46 |
support = describe_model_support(model)
|
| 47 |
|
| 48 |
assert support.supports_attribution is True
|
| 49 |
+
assert support.layer_path == "model.h_stack"
|
| 50 |
assert support.attention_attr == "attention"
|