refactor: added returns

566f8cbe · niklasfranz · 6aeb5c6e · 566f8cbe
Commit 566f8cbe authored 1 year ago by niklasfranz
--- a/src/Dataset.py
+++ b/src/Dataset.py
@@ -122,7 +122,7 @@ class Dataset:
        is_competitive_col = self.get_is_competitive_col(dataframe)
        return is_competitive_col

-    def treat_outliers(self, df, colname):
+    def treat_outliers(self, df, colname) -> pd.DataFrame:
        q = df[colname].quantile(0.99)
        return df[df[colname] < q]