Infer signature fix - no longer needed

Author: rayryeng

src/eda/conda.yml

@@ -8,7 +8,7 @@ dependencies:
   - matplotlib=3.8.2
   - pandas=2.1.3
   - pip=23.3.1
-  - scikit-learn=1.3.2
+  - scikit-learn=1.5.2
   - jupyterlab=4.1.3
   - pip:
       - mlflow==2.8.1

src/train_random_forest/run.py

@@ -17,7 +17,7 @@
 from sklearn.feature_extraction.text import TfidfVectorizer
 from sklearn.impute import SimpleImputer
 from sklearn.model_selection import train_test_split
-from sklearn.preprocessing import OrdinalEncoder, FunctionTransformer
+from sklearn.preprocessing import OrdinalEncoder, FunctionTransformer, OneHotEncoder
 
 import wandb
 from sklearn.ensemble import RandomForestRegressor
@@ -95,10 +95,8 @@ def go(args):
     ######################################
     # Save the sk_pipe pipeline as a mlflow.sklearn model in the directory "random_forest_dir"
     # HINT: use mlflow.sklearn.save_model
-    signature = mlflow.models.infer_signature(X_val, y_pred)
     mlflow.sklearn.save_model(
         # YOUR CODE HERE
-        signature = signature,
         input_example = X_train.iloc[:5]
     )
     ######################################