Updaed pickle to dill (#3458)

SeldonIO · Aug 10, 2021 · bbfa54f · bbfa54f
1 parent e580c2a
commit bbfa54f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/components/alibi-detect-server/adserver/cm_model.py b/components/alibi-detect-server/adserver/cm_model.py
@@ -1,7 +1,7 @@
 import json
 from typing import List, Dict, Optional, Union
 import logging
-import pickle
+import dill
 import os
 from adserver.constants import (
     REQUEST_ID_HEADER_NAME,
@@ -70,7 +70,7 @@ def load(self):
         """
         if "/" in self.storage_uri:
             model_folder = download_model(self.storage_uri)
-            self.model = pickle.load(
+            self.model = dill.load(
                 open(os.path.join(model_folder, "meta.pickle"), "rb")
             )
         else:

diff --git a/components/alibi-detect-server/setup.py b/components/alibi-detect-server/setup.py
@@ -23,6 +23,7 @@
         "google-cloud-storage==1.31.2",
         "seldon_core",
         "transformers>=4.0.0",
+        "dill==0.3.2",
     ],
     tests_require=tests_require,
     extras_require={"test": tests_require},