( self, batch: Dict[str, torch.tensor], )
source not stored for this graph (policy: none)
nothing calls this directly
no test coverage detected