Export FiniteDiscrete in tfp/d/__init__ and sort __all__ in same file.

PiperOrigin-RevId: 387829228

Author: csuter

tensorflow_probability/python/distributions/__init__.py

@@ -166,10 +166,6 @@
 # pylint: enable=line-too-long
 
 __all__ = [
-    'FULLY_REPARAMETERIZED',
-    'NOT_REPARAMETERIZED',
-    'ReparameterizationType',
-    'Distribution',
     'Autoregressive',
     'BatchBroadcast',
     'BatchReshape',
@@ -185,34 +181,38 @@
     'Chi',
     'Chi2',
     'CholeskyLKJ',
+    'DeterminantalPointProcess',
     'Deterministic',
+    'Dirichlet',
+    'DirichletMultinomial',
+    'Distribution',
     'DoublesidedMaxwell',
-    'VectorDeterministic',
-    'DeterminantalPointProcess',
     'Empirical',
-    'ExponentiallyModifiedGaussian',
     'ExpGamma',
     'ExpInverseGamma',
     'Exponential',
-    'VectorExponentialDiag',
+    'ExponentiallyModifiedGaussian',
+    'ExpRelaxedOneHotCategorical',
+    'FiniteDiscrete',
+    'FULLY_REPARAMETERIZED',
     'Gamma',
     'GammaGamma',
-    'InverseGaussian',
+    'GaussianProcess',
+    'GaussianProcessRegressionModel',
+    'GeneralizedExtremeValue',
     'GeneralizedNormal',
     'GeneralizedPareto',
     'Geometric',
-    'GaussianProcess',
-    'GaussianProcessRegressionModel',
-    'VariationalGaussianProcess',
     'Gumbel',
-    'GeneralizedExtremeValue',
     'HalfCauchy',
     'HalfNormal',
     'HalfStudentT',
     'HiddenMarkovModel',
     'Horseshoe',
     'Independent',
+    'independent_joint_distribution_from_structure',
     'InverseGamma',
+    'InverseGaussian',
     'JohnsonSU',
     'JointDistribution',
     'JointDistributionCoroutine',
@@ -221,25 +221,55 @@
     'JointDistributionNamedAutoBatched',
     'JointDistributionSequential',
     'JointDistributionSequentialAutoBatched',
+    'kl_divergence',
     'Kumaraswamy',
     'LambertWDistribution',
     'LambertWNormal',
     'Laplace',
     'LinearGaussianStateSpaceModel',
     'LKJ',
     'Logistic',
+    'LogitNormal',
     'LogLogistic',
     'LogNormal',
-    'LogitNormal',
     'MarkovChain',
+    'Masked',
+    'MatrixNormalLinearOperator',
+    'MatrixTLinearOperator',
+    'Mixture',
+    'MixtureSameFamily',
     'Moyal',
+    'Multinomial',
+    'MultivariateNormalDiag',
+    'MultivariateNormalDiagPlusLowRank',
+    'MultivariateNormalFullCovariance',
+    'MultivariateNormalLinearOperator',
+    'MultivariateNormalTriL',
+    'MultivariateStudentTLinearOperator',
+    'mvn_conjugate_linear_update',
     'NegativeBinomial',
     'Normal',
+    'normal_conjugates_known_scale_posterior',
+    'normal_conjugates_known_scale_predictive',
     'NormalInverseGaussian',
+    'NOT_REPARAMETERIZED',
+    'OneHotCategorical',
+    'OrderedLogistic',
+    'Pareto',
+    'PERT',
     'PixelCNN',
+    'PlackettLuce',
     'Poisson',
     'PoissonLogNormalQuadratureCompound',
+    'PowerSpherical',
     'ProbitBernoulli',
+    'quadrature_scheme_lognormal_gauss_hermite',
+    'quadrature_scheme_lognormal_quantiles',
+    'QuantizedDistribution',
+    'RegisterKL',
+    'RelaxedBernoulli',
+    'RelaxedOneHotCategorical',
+    'ReparameterizationType',
     'Sample',
     'SigmoidBeta',
     'SinhArcsinh',
@@ -248,47 +278,18 @@
     'StoppingRatioLogistic',
     'StudentT',
     'StudentTProcess',
+    'TransformedDistribution',
     'Triangular',
     'TruncatedCauchy',
     'TruncatedNormal',
     'Uniform',
-    'Masked',
-    'MatrixNormalLinearOperator',
-    'MatrixTLinearOperator',
-    'MultivariateNormalDiag',
-    'MultivariateNormalFullCovariance',
-    'MultivariateNormalLinearOperator',
-    'MultivariateNormalTriL',
-    'MultivariateNormalDiagPlusLowRank',
-    'MultivariateStudentTLinearOperator',
-    'Dirichlet',
-    'DirichletMultinomial',
-    'Multinomial',
+    'VariationalGaussianProcess',
+    'VectorDeterministic',
+    'VectorExponentialDiag',
     'VonMises',
     'VonMisesFisher',
     'Weibull',
     'WishartLinearOperator',
     'WishartTriL',
-    'TransformedDistribution',
-    'QuantizedDistribution',
-    'Mixture',
-    'MixtureSameFamily',
-    'ExpRelaxedOneHotCategorical',
-    'OneHotCategorical',
-    'OrderedLogistic',
-    'Pareto',
-    'PERT',
-    'PlackettLuce',
-    'PowerSpherical',
-    'RelaxedBernoulli',
-    'RelaxedOneHotCategorical',
     'Zipf',
-    'kl_divergence',
-    'RegisterKL',
-    'independent_joint_distribution_from_structure',
-    'mvn_conjugate_linear_update',
-    'normal_conjugates_known_scale_posterior',
-    'normal_conjugates_known_scale_predictive',
-    'quadrature_scheme_lognormal_gauss_hermite',
-    'quadrature_scheme_lognormal_quantiles',
 ]