aboutsummaryrefslogtreecommitdiff
path: root/r_qtl/r_qtl2.py
diff options
context:
space:
mode:
Diffstat (limited to 'r_qtl/r_qtl2.py')
-rw-r--r--r_qtl/r_qtl2.py18
1 files changed, 7 insertions, 11 deletions
diff --git a/r_qtl/r_qtl2.py b/r_qtl/r_qtl2.py
index b2a7acf..b688404 100644
--- a/r_qtl/r_qtl2.py
+++ b/r_qtl/r_qtl2.py
@@ -8,14 +8,10 @@ from typing import Iterator, Iterable, Callable
import yaml
-from quality_control.parsing import take
+from functional_tools import take, chain
from r_qtl.errors import InvalidFormat
-def thread_op(value, *functions):
- """Thread the `value` through the sequence of `functions`."""
- return reduce(lambda result, func: func(result), functions, value)
-
def control_data(zfile: ZipFile) -> dict:
"""Retrieve the control file from the zip file info."""
files = tuple(filename
@@ -140,7 +136,7 @@ def make_process_data_geno(cdata) -> tuple[
return val
def __non_transposed__(row: dict) -> dict:
return {
- key: thread_op(value, replace_genotype_codes, replace_na_strings)
+ key: chain(value, replace_genotype_codes, replace_na_strings)
for key,value in row.items()
}
def __transposed__(id_key: str,
@@ -149,7 +145,7 @@ def make_process_data_geno(cdata) -> tuple[
return tuple(
dict(zip(
[id_key, vals[0]],
- (thread_op(item, replace_genotype_codes, replace_na_strings)
+ (chain(item, replace_genotype_codes, replace_na_strings)
for item in items)))
for items in zip(ids, vals[1:]))
return (__non_transposed__, __transposed__)
@@ -179,7 +175,7 @@ def make_process_data_covar(cdata) -> tuple[
rep_cross_info = partial(replace_cross_info, cdata=cdata)
def non_transposed(row: dict) -> dict:
return {
- key: thread_op(value, rep_sex_info, rep_cross_info)
+ key: chain(value, rep_sex_info, rep_cross_info)
for key,value in row.items()
}
def transposed(id_key: str,
@@ -188,7 +184,7 @@ def make_process_data_covar(cdata) -> tuple[
return tuple(
dict(zip(
[id_key, vals[0]],
- (thread_op(item, rep_sex_info, rep_cross_info)
+ (chain(item, rep_sex_info, rep_cross_info)
for item in items)))
for items in zip(ids, vals[1:]))
return (non_transposed, transposed)
@@ -245,7 +241,7 @@ def cross_information(zfile: ZipFile, cdata: dict) -> Iterator[dict]:
new_cdata,
*make_process_data_covar(cdata)):
yield {
- key: thread_op(value, partial(replace_cross_info, cdata=cdata))
+ key: chain(value, partial(replace_cross_info, cdata=cdata))
for key, value in row.items() if key not in sex_fields}
def sex_information(zfile: ZipFile, cdata: dict) -> Iterator[dict]:
@@ -263,7 +259,7 @@ def sex_information(zfile: ZipFile, cdata: dict) -> Iterator[dict]:
new_cdata,
*make_process_data_covar(cdata)):
yield {
- key: thread_op(value, partial(replace_sex_info, cdata=cdata))
+ key: chain(value, partial(replace_sex_info, cdata=cdata))
for key, value in row.items() if key not in ci_fields}
def validate_bundle(zfile: ZipFile):