Change test to assert 4D hook shape for attention pattern

Author: degenfabian

tests/integration/model_bridge/test_bridge_integration.py

@@ -315,8 +315,8 @@ def capture_pattern_hook(tensor, hook):
 
         # Verify the shape is (n_heads, pos, pos) - attention patterns should not have batch dimension
         assert (
-            len(pattern_tensor.shape) == 3
-        ), f"Pattern tensor should be 3D, got {len(pattern_tensor.shape)}D"
+            len(pattern_tensor.shape) == 4
+        ), f"Pattern tensor should be 4D, got {len(pattern_tensor.shape)}D"
 
         n_heads_dim, pos_q_dim, pos_k_dim = pattern_tensor.shape