Update functions.py

functions.py

@@ -37,28 +37,6 @@ def decode_features(df, feature_view):
     return df_res
 
 
-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"
-    list_of_files = [os.path.join(dirpath,filename) for dirpath, _, filenames \
-                     in os.walk('.') for filename in filenames if filename == TARGET_FILE]
-
-    if list_of_files:
-        model_path = list_of_files[0]
-        model = joblib.load(model_path)
-    else:
-        if not os.path.exists(TARGET_FILE):
-            mr = project.get_model_registry()
-            # get best model based on custom metrics
-            model = mr.get_best_model(model_name,
-                                      evaluation_metric,
-                                      sort_metrics_by)
-            model_dir = model.download()
-            model = joblib.load(model_dir + "/model.pkl")
-
-    return model
-
 
 def get_air_json(AIR_QUALITY_API_KEY):
     return requests.get(f'https://api.waqi.info/feed/Helsinki/?token={AIR_QUALITY_API_KEY}').json()['data']