add __all__

aadya940 · Jul 21, 2024 · 66a0ab2 · 66a0ab2
1 parent 84e56a7
commit 66a0ab2
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 0 deletions.
diff --git a/chainopy/markov_chain.py b/chainopy/markov_chain.py
@@ -16,6 +16,8 @@
     _is_communicating,
 )
 
+__all__ = ["MarkovChain"]
+
 
 class MarkovChain:
     """

diff --git a/chainopy/markov_switching.py b/chainopy/markov_switching.py
@@ -4,6 +4,8 @@
 
 from .markov_chain import MarkovChain
 
+__all__ = ["MarkovSwitchingModel"]
+
 
 class MarkovSwitchingModel:
     """

diff --git a/chainopy/nn.py b/chainopy/nn.py
@@ -9,6 +9,8 @@
 from .markov_chain import MarkovChain
 from ._backend import _learn_matrix
 
+__all__ = ["MarkovChainNeuralNetwork", "divergance_analysis"]
+
 
 class MarkovChainNeuralNetwork(nn.Module):
     """
-Original file line number
+Diff line change
@@ Expand Up / @@ -16,6 +16,8 @@ @@
         _is_communicating,
     )
+    __all__ = ["MarkovChain"]
     class MarkovChain:
         """
@@ Expand Down @@