Merge pull request #277 from cvxgrp/276-expose-state

exposed state
cvxgrp · Jan 2, 2024 · 1d91216 · 1d91216
2 parents 65f3acb + f308187
commit 1d91216
Show file tree

Hide file tree

Showing 7 changed files with 193 additions and 215 deletions.
diff --git a/cvx/simulator/__init__.py b/cvx/simulator/__init__.py
@@ -12,9 +12,7 @@
 #    See the License for the specific language governing permissions and
 #    limitations under the License.
 
-# from cvx.simulator.futures.builder import FuturesBuilder
-# from cvx.simulator.futures.portfolio import FuturesPortfolio
-
 from cvx.simulator.builder import Builder
 from cvx.simulator.portfolio import Portfolio
+from cvx.simulator.state import State
 from cvx.simulator.utils.interpolation import interpolate
diff --git a/cvx/simulator/_state/__init__.py b/cvx/simulator/_state/__init__.py
diff --git a/cvx/simulator/builder.py b/cvx/simulator/builder.py
@@ -20,11 +20,10 @@
 import pandas as pd
 
 from cvx.simulator.portfolio import Portfolio
+from cvx.simulator.state import State
 from cvx.simulator.utils.interpolation import valid
 from cvx.simulator.utils.rescale import returns2prices
 
-from ._state.state import State
-
 
 @dataclass
 class Builder:

diff --git a/cvx/simulator/_state/state.py → cvx/simulator/state.py b/cvx/simulator/_state/state.py → cvx/simulator/state.py
diff --git a/poetry.lock b/poetry.lock
diff --git a/tests/test__state/__init__.py b/tests/test__state/__init__.py
diff --git a/tests/test__state/test_state.py → tests/test_state.py b/tests/test__state/test_state.py → tests/test_state.py
@@ -2,7 +2,7 @@
 import pandas as pd
 import pytest
 
-from cvx.simulator._state.state import State
+from cvx.simulator import State
 
 
 @pytest.fixture()