fix ring distributed test (#2380)

This commit is contained in:
Awni Hannun
2025-07-16 11:25:24 -07:00
committed by GitHub
parent 30571e2326
commit f409b229a4
2 changed files with 5 additions and 2 deletions

View File

@@ -4,6 +4,7 @@ import unittest
import mlx.core as mx
import mlx_distributed_tests
import mlx_tests
class TestRingDistributed(mlx_distributed_tests.MLXDistributedCommonTestCase):