From d9679529f494c1c3035e38c9a2fe424f01ca7802 Mon Sep 17 00:00:00 2001
From: Maximilian Schmidt <max.schmidt@fz-juelich.de>
Date: Wed, 6 Jun 2018 09:38:42 +0900
Subject: [PATCH] Remove data_wrapper

---
 figures/SchueckerSchmidt2017/utils.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/figures/SchueckerSchmidt2017/utils.py b/figures/SchueckerSchmidt2017/utils.py
index 506fc4f..556bd57 100644
--- a/figures/SchueckerSchmidt2017/utils.py
+++ b/figures/SchueckerSchmidt2017/utils.py
@@ -1,5 +1,4 @@
 import copy
-import data_wrapper.data_wrapper as dw
 import multiprocessing as mp
 import numpy as np
 import os
@@ -90,7 +89,8 @@ def save_iteration(step, data):
         os.mkdir(data_dir)
     except FileExistsError:
         pass
-    dw.save('iteration_{}'.format(step), data)
+    for key in ['parameters', 'K_prime', 'results']:
+        np.save('iteration_{}/{}.npy'.format(step, key), data[key])
 
 
 def load_iteration(step):
-- 
GitLab