Replaced pickle with json for security reasons

deepof/train_utils.py

@@ -17,9 +17,9 @@ from tensorboard.plugins.hparams import api as hp
 from typing import Tuple, Union, Any, List
 import deepof.hypermodels
 import deepof.model_utils
+import json
 import numpy as np
 import os
-import pickle
 import tensorflow as tf
 
 # Ignore warning with no downstream effect
@@ -53,11 +53,11 @@ def load_treatments(train_path):
         with open(
             os.path.join(
                 train_path,
-                [i for i in os.listdir(train_path) if i.endswith(".pkl")][0],
+                [i for i in os.listdir(train_path) if i.endswith(".json")][0],
             ),
-            "rb",
+            "r",
         ) as handle:
-            treatment_dict = pickle.load(handle)
+            treatment_dict = json.load(handle)
     except IndexError:
         treatment_dict = None
 

tests/test_examples/test_single_topview/Others/test_hparams.json

+{"units_conv": 256, "units_lstm": 256, "units_dense2": 64, "dropout_rate": 0.25, "encoding": 16, "learning_rate": 0.001}
\ No newline at end of file

tests/test_train_utils.py

@@ -22,7 +22,7 @@ import tensorflow as tf
 
 
 def test_load_treatments():
-    assert deepof.train_utils.load_treatments(".") is None
+    assert deepof.train_utils.load_treatments("tests") is None
     assert isinstance(
         deepof.train_utils.load_treatments(
             os.path.join("tests", "test_examples", "test_single_topview", "Others")