From 220223af48a37b632fe5b319997dfa759a4b213b Mon Sep 17 00:00:00 2001
From: edknv <edwardk@nvidia.com>
Date: Sat, 6 Jan 2024 19:08:31 -0800
Subject: [PATCH] trigger horovod test

---
 merlin/models/tf/distributed/backend.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/merlin/models/tf/distributed/backend.py b/merlin/models/tf/distributed/backend.py
index 78445d3b44..91f9d7707c 100644
--- a/merlin/models/tf/distributed/backend.py
+++ b/merlin/models/tf/distributed/backend.py
@@ -22,7 +22,7 @@
 
 if HAS_GPU:
     try:
-        from sparse_operation_kit import experiment as sok  # noqa: F401
+        import sparse_operation_kit as sok  # noqa: F401
 
         sok_installed = True
     except (ImportError, tf.errors.NotFoundError):