|
24 | 24 | iflogger = logging.getLogger("nipype.interface")
|
25 | 25 |
|
26 | 26 |
|
| 27 | +def _read_pickle(fname): |
| 28 | +with open(fname, 'rb') as f: |
| 29 | +return pickle.load(f) |
| 30 | + |
| 31 | + |
27 | 32 | def read_unknown_ntwk(ntwk):
|
28 | 33 | if not isinstance(ntwk, nx.classes.graph.Graph):
|
29 | 34 | _, _, ext = split_filename(ntwk)
|
30 | 35 | if ext == ".pck":
|
31 |
| -ntwk = nx.read_gpickle(ntwk) |
| 36 | +ntwk = _read_pickle(ntwk) |
32 | 37 | elif ext == ".graphml":
|
33 | 38 | ntwk = nx.read_graphml(ntwk)
|
34 | 39 | return ntwk
|
@@ -121,7 +126,7 @@ def average_networks(in_files, ntwk_res_file, group_id):
|
121 | 126 | counting_ntwk = ntwk.copy()
|
122 | 127 | # Sums all the relevant variables
|
123 | 128 | for index, subject in enumerate(in_files):
|
124 |
| -tmp = nx.read_gpickle(subject) |
| 129 | +tmp = _read_pickle(subject) |
125 | 130 | iflogger.info("File %s has %i edges", subject, tmp.number_of_edges())
|
126 | 131 | edges = list(tmp.edges())
|
127 | 132 | for edge in edges:
|
@@ -461,7 +466,7 @@ def _run_interface(self, runtime):
|
461 | 466 | edgentwks = list()
|
462 | 467 | kntwks = list()
|
463 | 468 | matlab = list()
|
464 |
| -ntwk = nx.read_gpickle(self.inputs.in_file) |
| 469 | +ntwk = _read_pickle(self.inputs.in_file) |
465 | 470 |
|
466 | 471 | # Each block computes, writes, and saves a measure
|
467 | 472 | # The names are then added to the output .pck file list
|
|
0 commit comments