diff options
Diffstat (limited to 'gn3/genodb.py')
| -rw-r--r-- | gn3/genodb.py | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/gn3/genodb.py b/gn3/genodb.py index abbc32e..7255d6e 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') -GenotypeMatrix = namedtuple('Matrix', 'db nrows ncols array transpose') +GenotypeMatrix = namedtuple('Matrix', 'array transpose') @contextmanager def open(path): @@ -54,8 +54,7 @@ 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 GenotypeMatrix(db, nrows, ncols, - np.reshape(np.frombuffer(read_optimized_blob[0 : nrows*ncols], + return GenotypeMatrix(np.reshape(np.frombuffer(read_optimized_blob[0 : nrows*ncols], dtype=np.uint8), (nrows, ncols)), np.reshape(np.frombuffer(read_optimized_blob[nrows*ncols :], |