trigger horovod test

NVIDIA-Merlin · Jan 7, 2024 · 9d98c38 · 9d98c38
1 parent 6bb5f00
commit 9d98c38
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/merlin/models/tf/distributed/backend.py b/merlin/models/tf/distributed/backend.py
@@ -10,7 +10,7 @@
 
 
 try:
-    import horovod.tensorflow.keras as hvd  # noqa: F401
+    import sparse_operation_kit as sok  # noqa: F401
 
     hvd_installed = True
 except ImportError: