Skip to content

Commit be8cc1d

Browse files
authored
Merge pull request #216 from worldcoin/dev
merge dev to main
2 parents ba11cc5 + 492e422 commit be8cc1d

3 files changed

Lines changed: 4 additions & 5 deletions

File tree

src/iris/nodes/geometry_estimation/lsq_ellipse_fit_with_refinement.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -47,10 +47,9 @@ def run(self, input_polygons: GeometryPolygons) -> Union[GeometryPolygons, None]
4747
if extrapolated_polygons is None:
4848
return None
4949

50+
extrapolated_pupil_copy = np.copy(extrapolated_polygons.pupil_array)
5051
for point in input_polygons.pupil_array:
51-
extrapolated_polygons.pupil_array[
52-
self._find_correspondence(point, extrapolated_polygons.pupil_array)
53-
] = point
52+
extrapolated_polygons.pupil_array[self._find_correspondence(point, extrapolated_pupil_copy)] = point
5453

5554
return extrapolated_polygons
5655

tests/e2e_tests/nodes/encoder/test_e2e_iris_encoder.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -27,5 +27,5 @@ def test_iris_encoder_constructor() -> None:
2727
assert len(result.iris_code_version) == len(expected_result.iris_code_version)
2828

2929
for i, (i_iris_code, i_mask_code) in enumerate(zip(result.iris_codes, result.mask_codes)):
30-
assert np.allclose(expected_result.iris_codes[i], i_iris_code, rtol=1e-05, atol=1e-07)
31-
assert np.allclose(expected_result.mask_codes[i], i_mask_code, rtol=1e-05, atol=1e-07)
30+
np.testing.assert_equal(expected_result.iris_codes[i], i_iris_code)
31+
np.testing.assert_equal(expected_result.mask_codes[i], i_mask_code)

0 commit comments

Comments
 (0)