Skip to content

Commit 5f18fe9

Browse files
committed
Remove unused filter code
1 parent c0dd73c commit 5f18fe9

File tree

1 file changed

+3
-18
lines changed

1 file changed

+3
-18
lines changed

vcztools/query.py

Lines changed: 3 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,6 @@
88
import zarr
99

1010
from vcztools import constants, retrieval
11-
from vcztools import filter as filter_mod
1211
from vcztools.utils import vcf_name_to_vcz_name
1312

1413

@@ -59,7 +58,7 @@ def __call__(self, *args, **kwargs):
5958

6059

6160
class QueryFormatGenerator:
62-
def __init__(self, root, query_format, filter_expr=None):
61+
def __init__(self, root, query_format):
6362
# TODO: pass in this metadata rather than root
6463
self.sample_ids = root["sample_id"][:].tolist()
6564
self.sample_count = len(self.sample_ids)
@@ -72,10 +71,7 @@ def __init__(self, root, query_format, filter_expr=None):
7271
assert isinstance(query_format, pp.ParseResults)
7372
parse_results = query_format
7473

75-
if filter_expr is None:
76-
filter_expr = filter_mod.FilterExpression()
77-
78-
self._generator = self._compose_generator(parse_results, filter_expr)
74+
self._generator = self._compose_generator(parse_results)
7975

8076
def __call__(self, *args, **kwargs):
8177
assert len(args) == 1
@@ -257,17 +253,9 @@ def generate(chunk_data):
257253

258254
return generate
259255

260-
def _compose_filter_generator(self, filter_expr):
261-
def generate(chunk_data):
262-
v_chunk_select = filter_expr.evaluate(chunk_data)
263-
yield from v_chunk_select
264-
265-
return generate
266-
267256
def _compose_generator(
268257
self,
269258
parse_results,
270-
filter_expr,
271259
) -> Callable:
272260
generators = (
273261
self._compose_element_generator(element) for element in parse_results
@@ -291,10 +279,7 @@ def write_query(
291279
exclude: str | None = None,
292280
):
293281
root = zarr.open(vcz, mode="r")
294-
filter_expr = filter_mod.FilterExpression(
295-
field_names=set(root), include=include, exclude=exclude
296-
)
297-
generator = QueryFormatGenerator(root, query_format, filter_expr)
282+
generator = QueryFormatGenerator(root, query_format)
298283

299284
for chunk_data in retrieval.variant_chunk_iter(
300285
root, include=include, exclude=exclude

0 commit comments

Comments
 (0)