Fix component test failures

- Fix PatchEmbed tests to use correct constructor arguments:
  - PatchEmbed(patch_size, in_chans, embed_dim) instead of (img_size, ...)
  - PatchEmbed3D expects (B, T, H, W, C) format input
- Fix get_2d_sincos_pos_embed test: function takes (embed_dim, grid_size)
- Fix allclose test: pos_embs returns numpy arrays, use np.allclose

Author: claude

tests/test_component_outputs.py

@@ -318,7 +318,7 @@ def test_patch_embed_2d_shape(self):
         in_chans = 3
         embed_dim = 768
 
-        patch_embed = PatchEmbed(img_size, patch_size, in_chans, embed_dim)
+        patch_embed = PatchEmbed(patch_size=patch_size, in_chans=in_chans, embed_dim=embed_dim)
         x = mx.random.normal((2, in_chans, img_size, img_size))
 
         output = patch_embed(x)
@@ -340,8 +340,9 @@ def test_patch_embed_3d_shape(self):
 
         num_frames = 8
 
-        patch_embed = PatchEmbed3D(img_size, patch_size, tubelet_size, in_chans, embed_dim)
-        x = mx.random.normal((1, in_chans, num_frames, img_size, img_size))
+        patch_embed = PatchEmbed3D(patch_size=patch_size, tubelet_size=tubelet_size, in_chans=in_chans, embed_dim=embed_dim)
+        # PatchEmbed3D expects (B, T, H, W, C) format
+        x = mx.random.normal((1, num_frames, img_size, img_size, in_chans))
 
         output = patch_embed(x)
         mx.eval(output)
@@ -365,7 +366,6 @@ def test_1d_sincos_shape(self):
         length = 100
 
         pos_embed = get_1d_sincos_pos_embed(embed_dim, length)
-        mx.eval(pos_embed)
 
         assert pos_embed.shape == (length, embed_dim)
 
@@ -374,8 +374,7 @@ def test_2d_sincos_shape(self):
         embed_dim = 768
         grid_size = 14
 
-        pos_embed = get_2d_sincos_pos_embed(embed_dim, grid_size, grid_size)
-        mx.eval(pos_embed)
+        pos_embed = get_2d_sincos_pos_embed(embed_dim, grid_size)
 
         num_patches = grid_size * grid_size
         assert pos_embed.shape == (num_patches, embed_dim)
@@ -387,10 +386,9 @@ def test_1d_sincos_different_lengths(self):
         pos_embed_100 = get_1d_sincos_pos_embed(embed_dim, 100)
         pos_embed_50 = get_1d_sincos_pos_embed(embed_dim, 50)
 
-        mx.eval(pos_embed_100, pos_embed_50)
-
-        # First 50 positions should be similar
-        assert mx.allclose(pos_embed_100[:50], pos_embed_50, atol=1e-5)
+        # First 50 positions should be similar (these are numpy arrays)
+        import numpy as np
+        assert np.allclose(pos_embed_100[:50], pos_embed_50, atol=1e-5)
 
 
 if __name__ == "__main__":