From ee3c177fb03c8115bc02c0b0b2464910e23881c2 Mon Sep 17 00:00:00 2001
From: Francesco Bruzzesi <42817048+FBruzzesi@users.noreply.github.com>
Date: Fri, 25 Aug 2023 23:17:44 +0200
Subject: [PATCH] Fix API docs rendering (#1041)

Nice, thanks!
---
 docs/api.rst | 34 +++++++++++++---------------------
 1 file changed, 13 insertions(+), 21 deletions(-)

diff --git a/docs/api.rst b/docs/api.rst
index 13457865..38b82fa5 100644
--- a/docs/api.rst
+++ b/docs/api.rst
@@ -62,33 +62,25 @@ This returns the matrix of log posterior probabilities for numerical stability.
 API Reference
 -------------
 
+Distributions
+=============
+
 .. automodule:: pomegranate.distributions
    :members: Bernoulli, Categorical, ConditionalCategorical, JointCategorical, DiracDelta, Exponential, Gamma, Normal, Poisson, StudentT, Uniform, ZeroInflated
 
-.. automodule:: pomegranate.bayes_classifier.BayesClassifier
-	:members:
-	:inherited-members:
+Models
+======
+
+.. autoclass:: pomegranate.bayes_classifier.BayesClassifier
 
-.. automodule:: pomegranate.gmm.GeneralMixtureModel
-	:members:
-	:inherited-members:
+.. autoclass:: pomegranate.gmm.GeneralMixtureModel
 
-.. automodule:: pomegranate.hmm.DenseHMM
-	:members:
-	:inherited-members:
+.. autoclass:: pomegranate.hmm.DenseHMM
 
-.. automodule:: pomegranate.hmm.SparseHMM
-	:members:
-	:inherited-members:
+.. autoclass:: pomegranate.hmm.SparseHMM
 
-.. automodule:: pomegranate.markov_chain.MarkovChain
-	:members:
-	:inherited-members:
+.. autoclass:: pomegranate.markov_chain.MarkovChain
 
-.. automodule:: pomegranate.bayesian_network.BayesianNetwork
-	:members:
-	:inherited-members:
+.. autoclass:: pomegranate.bayesian_network.BayesianNetwork
 
-.. automodule:: pomegranate.factor_graph.FactorGraph
-	:members:
-	:inherited-members:
\ No newline at end of file
+.. autoclass:: pomegranate.factor_graph.FactorGraph
\ No newline at end of file