bellman.environments.mixins.BatchSizeUpdaterMixin

class bellman.environments.mixins.BatchSizeUpdaterMixin[source]

Bases: abc.ABC

Mixin for environment model components that use the batch size as part of their internal state.

Methods

update_batch_size

type batch_size

int
abstract update_batch_size(batch_size)[source]
Parameters

batch_size (int) – New value for batch size

Return type

None