fixup! Split data using k-fold

src/preprocessing.py

@@ -31,7 +31,7 @@ def encode_columns(df):
 
 
 def split_train_target(df):
-    train_data = df.drop(labels="Severity")
+    train_data = df.drop(columns=["Severity"])
     target_data = df["Severity"]
     return train_data, target_data