From 6ad56402115f4c40a23931ea6836a54d0f84288f Mon Sep 17 00:00:00 2001 From: Emma Date: Mon, 18 Dec 2023 17:50:00 +0800 Subject: [PATCH] Update SOK import in backend.py --- 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 4e754106ed..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 sparse_operation_kit as sok # noqa: F401 + import sparse_operation_kit as sok # noqa: F401 sok_installed = True except (ImportError, tf.errors.NotFoundError):