From 75b08a39f6b845e8c7ee5d3066250012f3154299 Mon Sep 17 00:00:00 2001
From: Jakub Fojt <jakub.fojt96@gmail.com>
Date: Thu, 18 Jan 2024 16:40:44 +0100
Subject: [PATCH] Read npy files

---
 .modules.kuba/.tools.sh | 16 +++++++++++-----
 1 file changed, 11 insertions(+), 5 deletions(-)

diff --git a/.modules.kuba/.tools.sh b/.modules.kuba/.tools.sh
index b8d7317..cfe684e 100644
--- a/.modules.kuba/.tools.sh
+++ b/.modules.kuba/.tools.sh
@@ -11,12 +11,18 @@ except ImportError:
 class Wrap:
     def __init__(self, fname):
         n = np.load(fname, allow_pickle=True)
-        files = n.files
         globs = globals()
-        print('Contains files', files)
-        for f in files:
-            setattr(self, f, n[f])
-            globs[f] = n[f]
+        if isinstance(n, np.lib.npyio.NpzFile):
+            files = n.files
+            print('Contains files', files)
+            for f in files:
+                setattr(self, f, n[f])
+                globs[f] = n[f]
+        else:
+            print('Loaded data of shape', n.shape)
+            setattr(self, 'data', n)
+            globs['data'] = n
+
 
 argv = argv[argv.index('--')+1:]
 archives = [Wrap(arg) for arg in argv]