From 061310003711bd4fbb0e732672b6e0beb4d48226 Mon Sep 17 00:00:00 2001 From: rjzamora Date: Fri, 26 Apr 2024 15:28:42 -0700 Subject: [PATCH] formatting --- merlin/systems/workflow/base.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/merlin/systems/workflow/base.py b/merlin/systems/workflow/base.py index dae3064e..54a63ef8 100644 --- a/merlin/systems/workflow/base.py +++ b/merlin/systems/workflow/base.py @@ -77,8 +77,10 @@ def _initialize_ops(self, workflow_node, visited=None, restrict=None): if visited is None: visited = set() - if workflow_node.op and hasattr(workflow_node.op, "inference_initialize") and ( - not restrict or workflow_node.op.label in restrict + if ( + workflow_node.op + and hasattr(workflow_node.op, "inference_initialize") + and (not restrict or workflow_node.op.label in restrict) ): inference_op = workflow_node.op.inference_initialize( workflow_node.selector, self.model_config