From 4d0d77ad005f0832a14c4bc050fad77ec727d476 Mon Sep 17 00:00:00 2001
From: Arun Isaac
Date: Fri, 17 Jun 2022 14:15:56 +0530
Subject: gn3: genodb: Rename Matrix named tuple to GenotypeMatrix.

* gn3/genodb.py (Matrix): Rename to GenotypeMatrix.
(matrix): Update invocation of Matrix.
---
 gn3/genodb.py | 14 ++++++++------
 1 file changed, 8 insertions(+), 6 deletions(-)

(limited to 'gn3')

diff --git a/gn3/genodb.py b/gn3/genodb.py
index 7ffaf8b..31ac76f 100644
--- a/gn3/genodb.py
+++ b/gn3/genodb.py
@@ -29,7 +29,7 @@ import numpy as np
 # pylint: disable=invalid-name,redefined-builtin
 
 GenotypeDatabase = namedtuple('GenotypeDatabase', 'txn hash_length')
-Matrix = namedtuple('Matrix', 'db nrows ncols array transpose')
+GenotypeMatrix = namedtuple('Matrix', 'db nrows ncols array transpose')
 
 @contextmanager
 def open(path):
@@ -54,11 +54,13 @@ def matrix(db):
     read_optimized_blob = get(db, get(db, b'current'))
     nrows = int.from_bytes(get_metadata(db, hash, 'nrows'), byteorder='little')
     ncols = int.from_bytes(get_metadata(db, hash, 'ncols'), byteorder='little')
-    return Matrix(db, nrows, ncols,
-                  np.reshape(np.frombuffer(read_optimized_blob[0 : nrows*ncols], dtype=np.uint8),
-                             (nrows, ncols)),
-                  np.reshape(np.frombuffer(read_optimized_blob[nrows*ncols :], dtype=np.uint8),
-                             (nrows, ncols)))
+    return GenotypeMatrix(db, nrows, ncols,
+                          np.reshape(np.frombuffer(read_optimized_blob[0 : nrows*ncols],
+                                                   dtype=np.uint8),
+                                     (nrows, ncols)),
+                          np.reshape(np.frombuffer(read_optimized_blob[nrows*ncols :],
+                                                   dtype=np.uint8),
+                                     (nrows, ncols)))
 
 def nparray(matrix):
     '''Get matrix as a 2D numpy array.'''
-- 
cgit v1.2.3