diff --git a/tiledb/core.cc b/tiledb/core.cc index bd6d15a803..c632b5a971 100644 --- a/tiledb/core.cc +++ b/tiledb/core.cc @@ -450,7 +450,7 @@ class PyQuery { py::object init_pyqc = cond.attr("init_query_condition"); try { - init_pyqc(uri_, attrs_); + init_pyqc(uri_, attrs_, ctx_); } catch (tiledb::TileDBError &e) { TPY_ERROR_LOC(e.what()); } catch (py::error_already_set &e) { diff --git a/tiledb/query_condition.py b/tiledb/query_condition.py index 7a73271667..b6c1a90ce6 100644 --- a/tiledb/query_condition.py +++ b/tiledb/query_condition.py @@ -130,8 +130,10 @@ def __post_init__(self): "(Is this an empty expression?)" ) - def init_query_condition(self, uri: str, query_attrs: List[str]): - qctree = QueryConditionTree(self.ctx, Array.load_typed(uri), query_attrs) + def init_query_condition(self, uri: str, query_attrs: List[str], ctx): + qctree = QueryConditionTree( + self.ctx, Array.load_typed(uri, ctx=ctx), query_attrs + ) self.c_obj = qctree.visit(self.tree.body) if not isinstance(self.c_obj, qc.PyQueryCondition):