Skip to content

Commit

Permalink
Update merlin/systems/workflow/base.py
Browse files Browse the repository at this point in the history
  • Loading branch information
rjzamora authored Jun 11, 2024
1 parent 681454a commit b4eecd4
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion merlin/systems/workflow/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -71,7 +71,7 @@ def __init__(self, workflow, output_dtypes, model_config, model_device):
# by the `NVT_CPP_OPS` environment variable while we sort out whether
# and how we want to use C++ implementations of NVTabular operators for
# performance optimization)
_nvt_cpp_ops = os.environ.get("NVT_CPP_OPS", "").split(",")
_nvt_cpp_ops = os.environ.get("NVT_CPP_OPS", "Categorify").split(",")
self._initialize_ops(self.workflow.output_node, restrict=_nvt_cpp_ops)

def _initialize_ops(self, workflow_node, visited=None, restrict=None):
Expand Down

0 comments on commit b4eecd4

Please sign in to comment.