
from . import *


from .bernoulli import Bernoulli
from .binomial import Binomial
from .categorical import Categorical
from .multinomial import Multinomial

from .poisson import Poisson

from .beta import Beta
from .beta import Complement
from .dirichlet import Dirichlet, DirichletConcentration

from .exponential import Exponential

from .gaussian import Gaussian, GaussianARD
from .wishart import Wishart
from .gamma import Gamma, GammaShape

from .gaussian import (GaussianGamma,
                       GaussianWishart)

from .gaussian_markov_chain import GaussianMarkovChain
from .gaussian_markov_chain import VaryingGaussianMarkovChain
from .gaussian_markov_chain import SwitchingGaussianMarkovChain

from .categorical_markov_chain import CategoricalMarkovChain

from .mixture import Mixture, MultiMixture
from .gate import Gate
from .concatenate import Concatenate

from .dot import Dot
from .dot import SumMultiply
from .add import Add
from .take import Take
from .gaussian import ConcatGaussian

from .logpdf import LogPDF

from .constant import Constant


from . import *


from .bernoulli import Bernoulli
from .binomial import Binomial
from .categorical import Categorical
from .multinomial import Multinomial

from .poisson import Poisson

from .beta import Beta
from .beta import Complement
from .dirichlet import Dirichlet, DirichletConcentration

from .exponential import Exponential

from .gaussian import Gaussian, GaussianARD
from .wishart import Wishart
from .gamma import Gamma, GammaShape

from .gaussian import (GaussianGamma,
                       GaussianWishart)

from .gaussian_markov_chain import GaussianMarkovChain
from .gaussian_markov_chain import VaryingGaussianMarkovChain
from .gaussian_markov_chain import SwitchingGaussianMarkovChain

from .categorical_markov_chain import CategoricalMarkovChain

from .mixture import Mixture, MultiMixture
from .gate import Gate
from .gate import Choose
from .concatenate import Concatenate

from .dot import Dot
from .dot import SumMultiply
from .add import Add
from .take import Take
from .gaussian import ConcatGaussian

from .logpdf import LogPDF

from .constant import Constant

Add Choose node to imported nodes
ENH: Add Choose node to imported nodes

  
  from . import *
  
  
  from .bernoulli import Bernoulli
  from .binomial import Binomial
  from .categorical import Categorical
  from .multinomial import Multinomial
  
  from .poisson import Poisson
  
  from .beta import Beta
  from .beta import Complement
  from .dirichlet import Dirichlet, DirichletConcentration
  
  from .exponential import Exponential
  
  from .gaussian import Gaussian, GaussianARD
  from .wishart import Wishart
  from .gamma import Gamma, GammaShape
  
  from .gaussian import (GaussianGamma,
                         GaussianWishart)
  
  from .gaussian_markov_chain import GaussianMarkovChain
  from .gaussian_markov_chain import VaryingGaussianMarkovChain
  from .gaussian_markov_chain import SwitchingGaussianMarkovChain
  
  from .categorical_markov_chain import CategoricalMarkovChain
  
  from .mixture import Mixture, MultiMixture
  from .gate import Gate
+ from .gate import Choose
  from .concatenate import Concatenate
  
  from .dot import Dot
  from .dot import SumMultiply
  from .add import Add
  from .take import Take
  from .gaussian import ConcatGaussian
  
  from .logpdf import LogPDF
  
  from .constant import Constant
  
Add Choose node to imported nodes
## Code Before:

from . import *


from .bernoulli import Bernoulli
from .binomial import Binomial
from .categorical import Categorical
from .multinomial import Multinomial

from .poisson import Poisson

from .beta import Beta
from .beta import Complement
from .dirichlet import Dirichlet, DirichletConcentration

from .exponential import Exponential

from .gaussian import Gaussian, GaussianARD
from .wishart import Wishart
from .gamma import Gamma, GammaShape

from .gaussian import (GaussianGamma,
                       GaussianWishart)

from .gaussian_markov_chain import GaussianMarkovChain
from .gaussian_markov_chain import VaryingGaussianMarkovChain
from .gaussian_markov_chain import SwitchingGaussianMarkovChain

from .categorical_markov_chain import CategoricalMarkovChain

from .mixture import Mixture, MultiMixture
from .gate import Gate
from .concatenate import Concatenate

from .dot import Dot
from .dot import SumMultiply
from .add import Add
from .take import Take
from .gaussian import ConcatGaussian

from .logpdf import LogPDF

from .constant import Constant

## Instruction:
Add Choose node to imported nodes
## Code After:

from . import *


from .bernoulli import Bernoulli
from .binomial import Binomial
from .categorical import Categorical
from .multinomial import Multinomial

from .poisson import Poisson

from .beta import Beta
from .beta import Complement
from .dirichlet import Dirichlet, DirichletConcentration

from .exponential import Exponential

from .gaussian import Gaussian, GaussianARD
from .wishart import Wishart
from .gamma import Gamma, GammaShape

from .gaussian import (GaussianGamma,
                       GaussianWishart)

from .gaussian_markov_chain import GaussianMarkovChain
from .gaussian_markov_chain import VaryingGaussianMarkovChain
from .gaussian_markov_chain import SwitchingGaussianMarkovChain

from .categorical_markov_chain import CategoricalMarkovChain

from .mixture import Mixture, MultiMixture
from .gate import Gate
from .gate import Choose
from .concatenate import Concatenate

from .dot import Dot
from .dot import SumMultiply
from .add import Add
from .take import Take
from .gaussian import ConcatGaussian

from .logpdf import LogPDF

from .constant import Constant
