Skip to content

Commit 92e8340

Browse files
Tristan DennisTristan Dennis
authored andcommitted
fix function definitions
1 parent 6730123 commit 92e8340

1 file changed

Lines changed: 18 additions & 56 deletions

File tree

malariagen_data/anoph/to_plink.py

Lines changed: 18 additions & 56 deletions
Original file line numberDiff line numberDiff line change
@@ -56,23 +56,26 @@ def _create_plink_outfile(
5656
):
5757
return f"{output_dir}/{region}.{n_snps}.{min_minor_ac}.{max_missing_an}.{thin_offset}"
5858

59-
def _biallelic_snps_to_plink(
59+
def biallelic_snps_to_plink(
6060
self,
61-
*,
6261
output_dir,
63-
region,
64-
n_snps,
65-
thin_offset,
66-
overwrite,
67-
sample_sets,
68-
sample_query,
69-
sample_indices,
70-
site_mask,
71-
min_minor_ac,
72-
max_missing_an,
73-
random_seed,
74-
inline_array,
75-
chunks,
62+
region: base_params.regions,
63+
n_snps: base_params.n_snps,
64+
thin_offset: base_params.thin_offset = 0,
65+
overwrite: bool = False,
66+
sample_sets: Optional[base_params.sample_sets] = None,
67+
sample_query: Optional[base_params.sample_query] = None,
68+
sample_indices: Optional[base_params.sample_indices] = None,
69+
site_mask: Optional[base_params.site_mask] = None,
70+
min_minor_ac: Optional[
71+
base_params.min_minor_ac
72+
] = pca_params.min_minor_ac_default,
73+
max_missing_an: Optional[
74+
base_params.max_missing_an
75+
] = pca_params.max_missing_an_default,
76+
random_seed: base_params.random_seed = 42,
77+
inline_array: base_params.inline_array = base_params.inline_array_default,
78+
chunks: base_params.chunks = base_params.native_chunks,
7679
):
7780
# Define output file
7881
plink_file_path = self._create_plink_outfile(
@@ -142,44 +145,3 @@ def _biallelic_snps_to_plink(
142145
)
143146

144147
return plink_file_path
145-
146-
def biallelic_snps_to_plink(
147-
self,
148-
output_dir,
149-
region: base_params.regions,
150-
n_snps: base_params.n_snps,
151-
thin_offset: base_params.thin_offset = 0,
152-
overwrite: bool = False,
153-
sample_sets: Optional[base_params.sample_sets] = None,
154-
sample_query: Optional[base_params.sample_query] = None,
155-
sample_indices: Optional[base_params.sample_indices] = None,
156-
site_mask: Optional[base_params.site_mask] = None,
157-
min_minor_ac: Optional[
158-
base_params.min_minor_ac
159-
] = pca_params.min_minor_ac_default,
160-
max_missing_an: Optional[
161-
base_params.max_missing_an
162-
] = pca_params.max_missing_an_default,
163-
random_seed: base_params.random_seed = 42,
164-
inline_array: base_params.inline_array = base_params.inline_array_default,
165-
chunks: base_params.chunks = base_params.native_chunks,
166-
):
167-
params = dict(
168-
output_dir=output_dir,
169-
region=region,
170-
n_snps=n_snps,
171-
thin_offset=thin_offset,
172-
overwrite=overwrite,
173-
sample_sets=sample_sets,
174-
sample_query=sample_query,
175-
sample_indices=sample_indices,
176-
site_mask=site_mask,
177-
min_minor_ac=min_minor_ac,
178-
max_missing_an=max_missing_an,
179-
random_seed=random_seed,
180-
)
181-
182-
filepath = self._biallelic_snps_to_plink(
183-
inline_array=inline_array, chunks=chunks, **params
184-
)
185-
return filepath

0 commit comments

Comments
 (0)