assume#
- class cuequivariance.assume#
assume
is a context manager or decorator to assume the irrep class and layout for a block of code or a function.Examples
As a context manager:
>>> with cue.assume(cue.SO3, cue.mul_ir): ... rep = cue.IrrepsAndLayout("2x1") >>> rep.irreps 2x1 >>> rep.layout (mul,irrep)
As a decorator:
>>> @cue.assume(cue.SO3, cue.mul_ir) ... def foo(): ... return cue.IrrepsAndLayout("2x1") >>> assert foo() == rep
- __init__(irrep_class=None, layout=None)#
- Parameters:
layout (IrrepsLayout | None)