Skip to content

Commit

Permalink
docstring cleanup: vincent's comments
Browse files Browse the repository at this point in the history
  • Loading branch information
purva-thakre committed Jul 21, 2024
1 parent 01f0386 commit babbbab
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions mitiq/lre/inference/multivariate_richardson.py
Original file line number Diff line number Diff line change
Expand Up @@ -46,12 +46,12 @@ def _create_variable_combinations(num_layers: int, degree: int) -> List[Any]:
variables = _get_variables(num_layers)
variable_combinations = []
for i in range(degree, -1, -1):
# Generate combinations for the current degree
# Ranges from max degree, max degree -1, ...0.
# Generate combinations for the current degree.
# Ranges from max degree, max degree -1, ..., 0.
combos = list(combinations_with_replacement(variables, i))
variable_combinations.append(combos)

# return a flattened list
# Return a flattened list.
return list(chain(*variable_combinations))


Expand All @@ -71,7 +71,7 @@ def full_monomial_basis(num_layers: int, degree: int) -> List[str]:
Returns:
Monomial basis terms required for multivariate
extrapolation upto max degree
extrapolation up to max degree
"""
variable_combinations = _create_variable_combinations(num_layers, degree)

Expand Down

0 comments on commit babbbab

Please sign in to comment.