project-codeflare · chcost · Jun 20, 2021 · Jun 20, 2021 · Jun 20, 2021 · Jun 20, 2021
diff --git a/codeflare/pipelines/Datamodel.py b/codeflare/pipelines/Datamodel.py
@@ -898,7 +898,7 @@ def has_single_estimator(self):
         if len(self.get_output_nodes()) > 1:
             return False
 
-        for node in self.__node_name_map__.keys():
+        for node in self.__node_name_map__.values():
             is_node_estimator = (node.get_node_input_type() == NodeInputType.OR)
             if is_node_estimator:
                 pre_nodes = self.get_pre_nodes(node)

diff --git a/main.py b/main.py
diff --git a/notebooks/Cross Product.ipynb b/notebooks/Cross Product.ipynb