diff --git a/nvtabular/__init__.py b/nvtabular/__init__.py index 71d597b6cc..1759cde183 100644 --- 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