From 3cdc23faf7ba1072abfcb56dea5b33fbf011d3dd Mon Sep 17 00:00:00 2001 From: Liam Zhang Date: Tue, 27 Jan 2026 08:08:47 +0000 Subject: [PATCH] Fix method name typo (#240) Signed-off-by: Liam Zhang --- .../cuequivariance/group_theory/equivariant_polynomial.py | 6 +++--- .../segmented_polynomials/segmented_polynomial.py | 4 ++-- .../segmented_polynomials/segmented_polynomial_test.py | 2 +- 3 files changed, 6 insertions(+), 6 deletions(-) diff --git a/cuequivariance/cuequivariance/group_theory/equivariant_polynomial.py b/cuequivariance/cuequivariance/group_theory/equivariant_polynomial.py index c6bc640..0b7ba6b 100644 --- a/cuequivariance/cuequivariance/group_theory/equivariant_polynomial.py +++ b/cuequivariance/cuequivariance/group_theory/equivariant_polynomial.py @@ -480,7 +480,7 @@ def filter_keep_operands(self, keep: list[bool]) -> EquivariantPolynomial: Use this method when you want to compute only a subset of the polynomial outputs and have control over which inputs to keep. For keeping all inputs (even if not used), use filter_keep_outputs. For automatically removing unused operands, - use filter_drop_unsued_operands. + use filter_drop_unused_operands. Args: keep (list of bool): List indicating which operands to keep. @@ -515,7 +515,7 @@ def filter_keep_outputs(self, keep: list[bool]) -> EquivariantPolynomial: assert len(keep) == self.num_outputs return self.filter_keep_operands([True] * self.num_inputs + keep) - def filter_drop_unsued_operands(self) -> EquivariantPolynomial: + def filter_drop_unused_operands(self) -> EquivariantPolynomial: """Remove all unused operands from the polynomial. Returns: @@ -534,7 +534,7 @@ def filter_drop_unsued_operands(self) -> EquivariantPolynomial: if used_flag ] - filtered_polynomial = self.polynomial.filter_drop_unsued_operands() + filtered_polynomial = self.polynomial.filter_drop_unused_operands() return EquivariantPolynomial( filtered_inputs, filtered_outputs, filtered_polynomial diff --git a/cuequivariance/cuequivariance/segmented_polynomials/segmented_polynomial.py b/cuequivariance/cuequivariance/segmented_polynomials/segmented_polynomial.py index 6f23fab..5238f8a 100644 --- a/cuequivariance/cuequivariance/segmented_polynomials/segmented_polynomial.py +++ b/cuequivariance/cuequivariance/segmented_polynomials/segmented_polynomial.py @@ -897,7 +897,7 @@ def filter_keep_operands(self, keep: list[bool]) -> SegmentedPolynomial: Use this method when you want to compute only a subset of the polynomial outputs and have control over which inputs to keep. For keeping all inputs (even if not used), use filter_keep_outputs. For automatically removing unused operands, - use filter_drop_unsued_operands. + use filter_drop_unused_operands. Args: keep (list of bool): List indicating which operands to keep. @@ -951,7 +951,7 @@ def filter_keep_outputs(self, keep: list[bool]) -> SegmentedPolynomial: assert len(keep) == self.num_outputs return self.filter_keep_operands([True] * self.num_inputs + keep) - def filter_drop_unsued_operands(self) -> SegmentedPolynomial: + def filter_drop_unused_operands(self) -> SegmentedPolynomial: """Remove all unused operands from the polynomial. Returns: diff --git a/cuequivariance/tests/segmented_polynomials/segmented_polynomial_test.py b/cuequivariance/tests/segmented_polynomials/segmented_polynomial_test.py index 177185f..9ac4b32 100644 --- a/cuequivariance/tests/segmented_polynomials/segmented_polynomial_test.py +++ b/cuequivariance/tests/segmented_polynomials/segmented_polynomial_test.py @@ -109,7 +109,7 @@ def test_remove_unused_buffers(): assert poly.used_operands() == [True, False, True, True] - cleaned = poly.filter_drop_unsued_operands() + cleaned = poly.filter_drop_unused_operands() assert cleaned.num_inputs == 2 and cleaned.num_outputs == 1 assert cleaned.used_operands() == [True, True, True]