oryx.core.state.function.FunctionModule

Encapsulates a staged function.

Inherits From: Module

Methods

call

View source

call_and_update

View source

flatten

View source

replace

View source

unflatten

View source

update

View source

variables

View source

__call__

View source

Emulates a regular function call.

A Module's dunder call will ensure state is updated after the function call by calling assign on the updated state before returning the output of the function.

Args
*args The arguments to the module.
**kwargs The keyword arguments to the module.

Returns
The output of the module.