bucky.model.state
Provide a class to hold the internal state vector to the compartment model (and track compartment indices).
buckyState
Class to manage the state of the bucky compartments (and their indices).
slice_to_cpu(s)
slice_to_cpu
Ensure the values of the slice aren’t cupy arrays to prevent an unsupported implict conversion in xp.r_.
bucky.model.state.
__getattribute__
Allow for . access to the compartment indices, otherwise return the ‘normal’ attribute.
__setattr__
Allow setting of compartments using . notation, otherwise default to normal attribute behavior.
init_S
Init the S compartment such that N=1.
state_shape
Return the shape of the internal state ndarray.
validate_state
Validate that the state is valid (finite, nonnegative, N=1).
zeros_like
Return a mostly shallow copy of self but with a zeroed out self.state.
bucky.model.rhs
bucky.model.vacc