Add ops to NVT import (#1834)

NVIDIA-Merlin · Jun 5, 2023 · 2ca6df0 · 2ca6df0
1 parent 669dc50
commit 2ca6df0
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/nvtabular/__init__.py b/nvtabular/__init__.py
@@ -21,8 +21,7 @@
 from merlin.core import dispatch, utils  # noqa
 from merlin.dag import ColumnSelector
 from merlin.schema import ColumnSchema, Schema
-from nvtabular import workflow  # noqa
-from nvtabular import _version
+from nvtabular import _version, ops, workflow  # noqa
 
 # suppress some warnings with cudf warning about column ordering with dlpack
 # and numba warning about deprecated environment variables