From 4a3aec22b3f92f0461ca7408d681f97b6fa4f0d1 Mon Sep 17 00:00:00 2001
From: erusseil <etiennerusseil63@gmail.com>
Date: Mon, 23 Oct 2023 16:08:35 +0200
Subject: [PATCH] Use joblib instead of pickle to load the model

---
 fink_science/agn/unit_examples.py | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/fink_science/agn/unit_examples.py b/fink_science/agn/unit_examples.py
index ce64384f..1699f2d4 100644
--- a/fink_science/agn/unit_examples.py
+++ b/fink_science/agn/unit_examples.py
@@ -1,11 +1,9 @@
 import pandas as pd
 import numpy as np
 import fink_science.agn.kernel as k
-import pickle
+import joblib
 
-
-with open(k.CLASSIFIER_ELASTICC, "rb") as f:
-    clf_unit = pickle.load(f)
+clf_unit = joblib.load(k.CLASSIFIER_ELASTICC)
 
 raw_ztf_unit = pd.DataFrame(
     {