Skip to content

Commit fdc7ea1

Browse files
committed
fix: allow noe values in new columns
1 parent 3ae5fc2 commit fdc7ea1

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/leopard_em/utils/ctf_utils.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -223,13 +223,13 @@ def _setup_ctf_kwargs_from_particle_stack(
223223
assert particle_stack["phase_shift"].nunique() == 1
224224
assert particle_stack["ctf_B_factor"].nunique() == 1
225225
assert (
226-
particle_stack["mag_matrix"].nunique() == 1
226+
particle_stack["mag_matrix"].nunique() <= 1
227227
), "mag_matrix must be the same across all particles"
228228
assert (
229-
particle_stack["even_zernikes"].nunique() == 1
229+
particle_stack["even_zernikes"].nunique() <= 1
230230
), "even_zernikes must be the same across all particles"
231231
assert (
232-
particle_stack["odd_zernikes"].nunique() == 1
232+
particle_stack["odd_zernikes"].nunique() <= 1
233233
), "odd_zernikes must be the same across all particles"
234234

235235
# Convert mag_matrix from list to 2x2 tensor if provided

0 commit comments

Comments
 (0)