@@ -720,7 +720,7 @@ def loadpkl(infile):
720
720
pklopen = gzip .open if infile .suffix == ".pklz" else open
721
721
722
722
t = time ()
723
- timeout = float (config [ "execution" ][ "job_finished_timeout" ] )
723
+ timeout = float (config . get ( "execution" , "job_finished_timeout" ) )
724
724
timed_out = True
725
725
while (time () - t ) < timeout :
726
726
try :
@@ -826,8 +826,8 @@ def read_stream(stream, logger=None, encoding=None):
826
826
827
827
828
828
def savepkl (filename , record , versioning = False , sync = False ):
829
- with open ( filename , "wb " ) as fobj :
830
- pkl_file = gzip . GzipFile ( fileobj = fobj ) if filename . endswith ( ".pklz " ) else fobj
829
+ pkl_open = gzip . open if filename . endswith ( ".pklz " ) else open
830
+ with pkl_open ( filename , "wb " ) as pkl_file :
831
831
if versioning :
832
832
from nipype import __version__ as version
833
833
@@ -837,9 +837,10 @@ def savepkl(filename, record, versioning=False, sync=False):
837
837
pkl_file .write ("\n " .encode ("utf-8" ))
838
838
839
839
pickle .dump (record , pkl_file )
840
+ fmlogger .info ("Finished saving: {}" .format (filename ))
840
841
if sync :
841
- fobj .flush ()
842
- os .fsync (fobj .fileno ())
842
+ pkl_file .flush ()
843
+ os .fsync (pkl_file .fileno ())
843
844
844
845
845
846
rst_levels = ["=" , "-" , "~" , "+" ]
0 commit comments