Check if estimator.get_model_container is needed for diffxpy. Do we change this in diffxpy or batchglm?