Add ops to NVT import

NVIDIA-Merlin · Jun 5, 2023 · a248e8c · a248e8c
1 parent bab6828
commit a248e8c
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