From 566f8cbe76cc3f0d6a78f5841e839953dcd31067 Mon Sep 17 00:00:00 2001
From: niklasfranz <nf.app@icloud.com>
Date: Wed, 12 Jul 2023 18:36:18 +0200
Subject: [PATCH] refactor: added returns
---
src/Dataset.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/Dataset.py b/src/Dataset.py
index 72cb721..312a6e2 100644
--- 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]
--
GitLab