MockBackend
In-memory Backend for tests.
Records every call to forward_backward_async /
forward_backward_custom_async / optim_step_async /
save_for_sampler / save_full_state /
snapshot_sampling_client for assertion. Returns predictable result
objects so the loop's metric / save / eval paths exercise end-to-end.
Attributes
attributefb_callslist[dict]= []attributefb_custom_callslist[dict]= []attributeoptim_callslist[tinker.AdamParams]= []attributesave_sampler_callslist[str]= []attributesave_state_callslist[str]= []attributefb_logprobfloat= -0.5attributeoptim_metricsdict[str, float]= {'optim/lr': 0.0001}Functions
func__init__(self, *, tokenizer=None, sampler_factory=None) -> NoneparamselfparamtokenizerAny= Noneparamsampler_factoryCallable[[str], MockSamplingClient] | None= NoneReturns
Nonefuncforward_backward_async(self, data, *, loss_fn, loss_fn_config=None) -> Anyparamselfparamdatalist[tinker.Datum]paramloss_fntinker.types.LossFnTypeparamloss_fn_configdict[str, Any] | None= NoneReturns
typing.Anyfuncforward_backward_custom_async(self, data, loss_fn) -> Anyparamselfparamdatalist[tinker.Datum]paramloss_fnLossCallableReturns
typing.Anyfuncoptim_step_async(self, adam) -> Anyparamselfparamadamtinker.AdamParamsReturns
typing.Anyfuncsave_for_sampler(self, name) -> strparamselfparamnamestrReturns
strfuncsave_full_state(self, name) -> strparamselfparamnamestrReturns
strfuncsnapshot_sampling_client(self, name=None) -> SamplingClientparamselfparamnamestr | None= NoneReturns
evsys_sdk.training.backend.SamplingClientfuncget_tokenizer(self) -> AnyparamselfReturns
typing.Any