From 725afb67b7fbe301620ceb8be415e41166d675ad Mon Sep 17 00:00:00 2001 From: qqiao Date: Sat, 16 Dec 2023 22:55:17 -0800 Subject: [PATCH] Fix sok import error in ut --- 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..4e754106ed 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 + from sparse_operation_kit import sparse_operation_kit as sok # noqa: F401 sok_installed = True except (ImportError, tf.errors.NotFoundError):