diff --git a/src/autora/experimentalist/novelty/__init__.py b/src/autora/experimentalist/novelty/__init__.py index 2e7a5e1..6f94a4d 100644 --- a/src/autora/experimentalist/novelty/__init__.py +++ b/src/autora/experimentalist/novelty/__init__.py @@ -71,7 +71,9 @@ def sample( new_conditions.drop("score", axis=1, inplace=True) if isinstance(condition_pool_copy, pd.DataFrame): - new_conditions = pd.DataFrame(new_conditions, columns=condition_pool_copy.columns) + new_conditions = pd.DataFrame( + new_conditions, columns=condition_pool_copy.columns + ) return new_conditions