Fix sok import error in ut

NVIDIA-Merlin · Dec 17, 2023 · 725afb6 · 725afb6
1 parent 80b086f
commit 725afb6
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
+ from sparse_operation_kit import sparse_operation_kit as sok # noqa: F401
 
  sok_installed = True
  except (ImportError, tf.errors.NotFoundError):