BART-ender commited on
Commit
2298e0a
·
verified ·
1 Parent(s): aca1293

test(core): update HRM test case to correct structure

Browse files
Files changed (1) hide show
  1. tests/test_model_support.py +2 -2
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=SimpleNamespace(layers=[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.layers"
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"