fixed typing bug

clsp/enumeration.py

@@ -112,7 +112,7 @@ def enumerate_term_vectors(
     non_terminals: tuple[S] | list[S],
     existing_terms: dict[S, set[Tree[T]]],
     nt_term: Optional[tuple[S, Tree[T]]] = None,
-) -> Iterable[Tree[T]]:
+) -> Iterable[tuple[Tree[T], ...]]:
     """Enumerate possible term vectors for a given list of non-terminals and existing terms. Use nt_term at least once (if given)."""
     if nt_term is None:
         yield from itertools.product(*(existing_terms[n] for n in non_terminals))
@@ -142,7 +142,7 @@ def generate_new_terms(
 
     if nt_old_term is None:
         all_args = list(enumerate_term_vectors(rule.args, existing_terms, None))
-        all_params = [
+        all_params: list[tuple[Tree[T], ...]] = [
             interleave(substitution)
             for param_terms in enumerate_term_vectors(param_nts, existing_terms, None)
             for substitution in (dict(zip(names, param_terms)),)