- [ ] Add keyword argument model=None - [ ] If self._layer is a ModelLayerBase, pass the model keyword into calls to .args and .kwargs