assert_mul_ir_dict

cuequivariance_jax.ir_dict.assert_mul_ir_dict(irreps, x)

Assert that a dict[Irrep, Array] matches the expected irreps structure.

Parameters:

• irreps (Irreps) – Expected irreps specification.

• x (dict[Irrep, Array]) – Dictionary mapping Irreps to arrays.

Raises:

AssertionError – If keys don’t match or array shapes are incorrect. Expected shape for each array is (…, multiplicity, irrep_dim).

Return type:

None