Use joblib instead of pickle to load the model

astrolabsoftware · Oct 23, 2023 · 4a3aec2 · 4a3aec2
1 parent 9dc1813
commit 4a3aec2
Showing 1 changed file with 2 additions and 4 deletions.
diff --git 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(
     {