Skip to content
This repository was archived by the owner on May 14, 2026. It is now read-only.

Commit a78c6d4

Browse files
[EQLMatch] fix and cleaning
1 parent 31ea32d commit a78c6d4

1 file changed

Lines changed: 3 additions & 8 deletions

File tree

  • src/krrood/entity_query_language

src/krrood/entity_query_language/match.py

Lines changed: 3 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -249,18 +249,12 @@ def infer_condition_between_attribute_and_assigned_value(
249249
isinstance(self.assigned_value, Match) and self.assigned_value.universal
250250
)
251251
):
252-
flat_attr = (
253-
flatten(self.attr) if not isinstance(self.attr, Flatten) else self.attr
254-
)
255-
condition = contains(self.assigned_variable, flat_attr)
252+
condition = contains(self.assigned_variable, flatten(self.attr))
256253
else:
257254
condition = self.attr == self.assigned_variable
258255

259256
if isinstance(self.assigned_value, Match) and self.assigned_value.existential:
260-
attr = (
261-
self.attr if not isinstance(self.attr, Flatten) else self.attr._child_
262-
)
263-
condition = exists(attr, condition)
257+
condition = exists(self.attr, condition)
264258

265259
return condition
266260

@@ -347,6 +341,7 @@ def _resolve(
347341
parent: Optional[Match] = None,
348342
):
349343
super()._resolve(variable, parent)
344+
variable = variable or self.variable
350345
if not self._var_:
351346
self._var_ = variable
352347

0 commit comments

Comments
 (0)