Also pickles the targets/protocols (#137)

* Also pickles the targets/protocols * add values properties --------- Co-authored-by: arnaudon <alexis.arnaudon@epfl.ch>
BlueBrain · Mar 23, 2023 · 44bb87d · 44bb87d
1 parent 4ed2ddb
commit 44bb87d
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/bluepyefe/extract.py b/bluepyefe/extract.py
@@ -987,6 +987,7 @@ def extract_efeatures(
             rheobase_strategy,
             rheobase_settings
         )
+
     efeatures, protocol_definitions, current = create_feature_protocol_files(
         cells,
         protocols,
@@ -995,10 +996,12 @@ def extract_efeatures(
         write_files=write_files,
         default_std_value=default_std_value
     )
+
     if pickle_cells:
         path_cells = pathlib.Path(output_directory)
         path_cells.mkdir(parents=True, exist_ok=True)
         pickle.dump(cells, open(path_cells / "cells.pkl", 'wb'))
+        pickle.dump(protocols, open(path_cells / "protocols.pkl", 'wb'))
 
     if plot:
         plot_all_recordings_efeatures(

diff --git a/bluepyefe/target.py b/bluepyefe/target.py
@@ -75,6 +75,11 @@ def __init__(
         self._files = []
         self._auto_thresholds = []
 
+    @property
+    def values(self):
+        """Return all values."""
+        return self._values
+
     @property
     def mean(self):
         """Average of the e-feature value at target"""