Fix sok import error in ut (#1232)

* Fix sok import error in ut * Update SOK import in backend.py
NVIDIA-Merlin · Jan 7, 2024 · 162f498 · 162f498
1 parent 80b086f
commit 162f498
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
@@ -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):