'BernoulliLogits' object has no attribute 'mask'

pookie · February 24, 2021, 10:26pm

Hello!
We are trying this tutorial:

pyro-ppl/numpyro/blob/master/notebooks/source/discrete_imputation.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "# Bayesian Imputation for Missing Values in Discrete Covariates\n",
    "\n",
    "Missing data is a very widespread problem in practical applications, both in covariates ('explanatory variables') and outcomes.\n",
    "When performing bayesian inference with MCMC, imputing discrete missing values is not possible using Hamiltonian Monte Carlo techniques.\n",
    "One way around this problem is to create a new model that enumerates the discrete variables and does inference over the new model, which, for a single discrete variable, is a mixture model. (see e.g. [Stan's user guide on Latent Discrete Parameters](https://mc-stan.org/docs/2_18/stan-users-guide/change-point-section.html))\n",
    "Enumerating the discrete latent sites requires some manual math work that can get tedious for complex models.\n",
    "Inference by automatic enumeration of discrete variables is implemented in numpyro and allows for a very convenient way of dealing with missing discrete data.\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],

This file has been truncated. show original

and we go this error message:
AttributeError: ‘BernoulliLogits’ object has no attribute ‘mask’

in the function impmodel().

Would you have suggestions on how to fix this?

Note: we use numpyro.infer import MCMC, NUTS.

fehiepsi · February 24, 2021, 10:28pm

Probably this is a bug. Could you open a github issue instead?

pookie · February 24, 2021, 10:43pm

@fehiepsi, Ok, a github issue was opened. Thank you.

pookie · February 25, 2021, 12:28am

Thank you very much @fehiepsi for quick answer in github. We updated Ubuntu, and funsor and it now works.