diff --git a/modules/deeplearning/srcnn_cld_frac.py b/modules/deeplearning/srcnn_cld_frac.py
index 26309e59e8322997b8757b2aba7c2ab7a6741046..d6d4658d52641b2988238bffbe323dbbcf38c67b 100644
--- a/modules/deeplearning/srcnn_cld_frac.py
+++ b/modules/deeplearning/srcnn_cld_frac.py
@@ -799,7 +799,8 @@ def run_restore_static(directory, ckpt_dir, out_file=None):
     nn = SRCNN()
     labels, preds = nn.run_restore(directory, ckpt_dir)
     if out_file is not None:
-        np.save(out_file, [np.squeeze(labels), preds.argmax(axis=3), preds[:,:,:,0], preds[:,:,:,1], preds[:,:,:,2]])
+        np.save(out_file,
+                [np.squeeze(labels), preds.argmax(axis=3), preds[:, :, :, 0], preds[:, :, :, 1], preds[:, :, :, 2]])
 
 
 def run_evaluate_static(in_file, out_file, ckpt_dir):