Просмотр исходного кода

Fix: remove NaN output of components. (#6079)

### What problem does this PR solve?

#6065

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
tags/v0.18.0
Kevin Hu 7 месяцев назад
Родитель
Сommit
12c3023a22
Аккаунт пользователя с таким Email не найден
1 измененных файлов: 4 добавлений и 4 удалений
  1. 4
    4
      agent/component/base.py

+ 4
- 4
agent/component/base.py Просмотреть файл

@@ -429,7 +429,7 @@ class ComponentBase(ABC):
if not isinstance(o, partial):
if not isinstance(o, pd.DataFrame):
if isinstance(o, list):
return self._param.output_var_name, pd.DataFrame(o)
return self._param.output_var_name, pd.DataFrame(o).dropna()
if o is None:
return self._param.output_var_name, pd.DataFrame()
return self._param.output_var_name, pd.DataFrame([{"content": str(o)}])
@@ -437,15 +437,15 @@ class ComponentBase(ABC):

if allow_partial or not isinstance(o, partial):
if not isinstance(o, partial) and not isinstance(o, pd.DataFrame):
return pd.DataFrame(o if isinstance(o, list) else [o])
return pd.DataFrame(o if isinstance(o, list) else [o]).dropna()
return self._param.output_var_name, o

outs = None
for oo in o():
if not isinstance(oo, pd.DataFrame):
outs = pd.DataFrame(oo if isinstance(oo, list) else [oo])
outs = pd.DataFrame(oo if isinstance(oo, list) else [oo]).dropna()
else:
outs = oo
outs = oo.dropna()
return self._param.output_var_name, outs

def reset(self):

Загрузка…
Отмена
Сохранить