Extract definition of function.

environment.yml

@@ -27,4 +27,4 @@ dependencies:
     - xgboost
     - pip:
         - biosppy
-        - cr_features
+        - cr_features>=0.2

src/features/all_cleaning_individual/straw/main.py

@@ -150,13 +150,15 @@ def straw_cleaning(sensor_data_files, provider):
 
     return features
 
-def impute(df, method='zero'):
 
-    def k_nearest(df):
+def k_nearest(df):
     pd.set_option('display.max_columns', None)
     imputer = KNNImputer(n_neighbors=3)
     return pd.DataFrame(imputer.fit_transform(df), columns=df.columns)
 
+
+def impute(df, method='zero'):
+
     return {
         'zero': df.fillna(0),
         'high_number': df.fillna(1500),
@@ -165,6 +167,7 @@ def impute(df, method='zero'):
         'knn': k_nearest(df) 
     }[method]
 
+
 def graph_bf_af(features, phase_name, plt_flag=False):
     if plt_flag:
         sns.set(rc={"figure.figsize":(16, 8)})

src/features/all_cleaning_overall/straw/main.py

@@ -245,12 +245,14 @@ def straw_cleaning(sensor_data_files, provider, target):
 
     return features
 
-def impute(df, method='zero'):
 
-    def k_nearest(df):
+def k_nearest(df):
     imputer = KNNImputer(n_neighbors=3)
     return pd.DataFrame(imputer.fit_transform(df), columns=df.columns)
 
+
+def impute(df, method='zero'):
+
     return {
         'zero': df.fillna(0),
         'high_number': df.fillna(1500),
@@ -259,6 +261,7 @@ def impute(df, method='zero'):
         'knn': k_nearest(df) 
     }[method]
 
+
 def graph_bf_af(features, phase_name, plt_flag=False):
     if plt_flag:
         sns.set(rc={"figure.figsize":(16, 8)})