jan rathfelder
10/06/2024, 2:13 PM--> warnings.warn(f'Found null values in {", ".join(cols_with_nulls)}.')
def _get_features_for_next_step(self, X_df=None):
new_x = self._update_features()
if X_df is not None:
n_series = len(self.uids)
h = X_df.shape[0] // n_series
rows = np.arange(self._h, X_df.shape[0], h)
X = ufp.take_rows(X_df, rows)
X = ufp.drop_index_if_pandas(X)
new_x = ufp.horizontal_concat([new_x, X])
if isinstance(new_x, pd.DataFrame):
nulls = new_x.isnull().any()
cols_with_nulls = nulls[nulls].index.tolist()
else:
nulls = new_x.select(pl.all().is_null().any())
cols_with_nulls = [k for k, v in nulls.to_dicts()[0].items() if v]
#if cols_with_nulls:
# warnings.warn(f'Found null values in {", ".join(cols_with_nulls)}.')
self._h += 1
new_x = new_x[self.features_order_]
if self.as_numpy:
new_x = ufp.to_numpy(new_x)
return new_x
José Morales
10/07/2024, 3:37 PMwarnings.filterwarnings("ignore", message=r"Found null values.*")
jan rathfelder
10/07/2024, 3:52 PMjan rathfelder
10/07/2024, 3:53 PM