Skip to content

Commit

Permalink
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Update merlin/systems/workflow/base.py
Browse files Browse the repository at this point in the history
rjzamora authored Apr 24, 2024

Verified

This commit was created on GitHub.com and signed with GitHub’s verified signature.
1 parent dafab0f commit 8813645
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion merlin/systems/workflow/base.py
Original file line number Diff line number Diff line change
@@ -71,7 +71,9 @@ def __init__(self, workflow, output_dtypes, model_config, model_device):
# performance optimization)
self._initialize_ops(self.workflow.output_node, restrict=["Categorify"])

def _initialize_ops(self, workflow_node, visited=None, restrict=False):
def _initialize_ops(self, workflow_node, visited=None, restrict=None):
restrict = restrict or []

if visited is None:
visited = set()

0 comments on commit 8813645

Please sign in to comment.