TODO: In pandas 2.0, mean is implemented for datetimes, but Dask returns None.
(df, col=None)
| 70 | |
| 71 | |
| 72 | def _drop_mean(df, col=None): |
| 73 | """TODO: In pandas 2.0, mean is implemented for datetimes, but Dask returns None.""" |
| 74 | if isinstance(df, pd.DataFrame): |
| 75 | df.at["mean", col] = np.nan |
| 76 | df.dropna(how="all", inplace=True) |
| 77 | elif isinstance(df, pd.Series): |
| 78 | df = df.drop(labels=["mean"], errors="ignore") |
| 79 | else: |
| 80 | raise NotImplementedError("Expected Series or DataFrame with mean") |
| 81 | return df |
| 82 | |
| 83 | |
| 84 | def test_Dataframe(): |
no test coverage detected