Update functions.py

functions.py

@@ -38,7 +38,7 @@ def decode_features(df, feature_view):
     return df_res
 
 
-def get_model(project, model_name, evaluation_metric, sort_metrics_by, version_name):
+def get_model(project, model_name, evaluation_metric, sort_metrics_by):
     """Retrieve desired model or download it from the Hopsworks Model Registry.
     In second case, it will be physically downloaded to this directory"""
     TARGET_FILE = "model.pkl"
@@ -54,8 +54,7 @@ def get_model(project, model_name, evaluation_metric, sort_metrics_by, version_n
             # get best model based on custom metrics
             model = mr.get_best_model(model_name,
                                       evaluation_metric,
-                                      sort_metrics_by,
-                                      version = version_name)
+                                      sort_metrics_by)
             model_dir = model.download()
             model = joblib.load(model_dir + "/model.pkl")