expose _markov_chain, _nn, _markov_switching through the public API

aadya940 · Jul 20, 2024 · 983e6de · 983e6de
1 parent 87bcde7
commit 983e6de
Show file tree

Hide file tree

Showing 8 changed files with 12 additions and 10 deletions.
diff --git a/chainopy/__init__.py b/chainopy/__init__.py
@@ -1,6 +1,6 @@
-from ._markov_chain import MarkovChain
-from ._nn import MarkovChainNeuralNetwork, divergance_analysis
-from ._markov_switching import MarkovSwitchingModel
+from .markov_chain import MarkovChain
+from .nn import MarkovChainNeuralNetwork, divergance_analysis
+from .markov_switching import MarkovSwitchingModel
 
 __all__ = [
     "MarkovChain",

diff --git a/chainopy/_markov_chain.py → chainopy/markov_chain.py b/chainopy/_markov_chain.py → chainopy/markov_chain.py
diff --git a/chainopy/_markov_switching.py → chainopy/markov_switching.py b/chainopy/_markov_switching.py → chainopy/markov_switching.py
@@ -2,7 +2,7 @@
 from typing import List, Tuple
 from statsmodels.tsa.ar_model import AutoReg
 
-from ._markov_chain import MarkovChain
+from .markov_chain import MarkovChain
 
 
 class MarkovSwitchingModel:

diff --git a/chainopy/_nn.py → chainopy/nn.py b/chainopy/_nn.py → chainopy/nn.py
@@ -6,7 +6,7 @@
 import warnings
 import random
 
-from ._markov_chain import MarkovChain
+from .markov_chain import MarkovChain
 from ._backend import _learn_matrix
 
 

diff --git a/chainopy/tests/test_markovchain.py b/chainopy/tests/test_markovchain.py
@@ -3,7 +3,7 @@
 import numpy as np
 import os
 
-from .._markov_chain import MarkovChain
+from ..markov_chain import MarkovChain
 
 
 def test_init():

diff --git a/chainopy/tests/test_nn.py b/chainopy/tests/test_nn.py
@@ -2,8 +2,8 @@
 import numpy as np
 import torch
 
-from .._nn import MarkovChainNeuralNetwork, divergance_analysis
-from .._markov_chain import MarkovChain
+from ..nn import MarkovChainNeuralNetwork, divergance_analysis
+from ..markov_chain import MarkovChain
 
 
 @pytest.fixture

diff --git a/chainopy/tests/test_switching.py b/chainopy/tests/test_switching.py
@@ -1,7 +1,7 @@
 import numpy as np
 import random
 
-from .._markov_switching import MarkovSwitchingModel
+from ..markov_switching import MarkovSwitchingModel
 
 
 def test_model_fit():

diff --git a/docs/source/index.rst b/docs/source/index.rst
@@ -12,7 +12,9 @@ Autosummary
 .. autosummary::
    :toctree: _autosummary
 
-   chainopy
+   chainopy.markov_chain
+   chainopy.markov_switching
+   chainopy.nn
 
 Readme
 ------