diff --git a/graph/component/base.py b/graph/component/base.py index 82e2f18c7..f0f6367f8 100644 --- a/graph/component/base.py +++ b/graph/component/base.py @@ -456,12 +456,17 @@ class ComponentBase(ABC): break break if self.component_name.lower().find("answer") >= 0: - if self.get_component_name(u) in ["relevant"]: continue - - else: upstream_outs.append(self._canvas.get_component(u)["obj"].output(allow_partial=False)[1]) + if self.get_component_name(u) in ["relevant"]: + continue + else: + o = self._canvas.get_component(u)["obj"].output(allow_partial=False)[1] + if o is not None: + upstream_outs.append(o) break - return pd.concat(upstream_outs, ignore_index=False) + if upstream_outs: + return pd.concat(upstream_outs, ignore_index=False) + return pd.DataFrame() def get_stream_input(self): reversed_cpnts = []