From 3de84ea9b7898d3df379b5be54e74c5c2d0c5ccd Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Javier=20Rodr=C3=ADguez=20Chatruc?= Date: Fri, 31 Jan 2020 11:27:23 -0300 Subject: [PATCH] Minor fix --- asset_backtester/__init__.py | 3 +-- asset_backtester/backtester.py | 2 +- asset_backtester/portfolio/portfolio.py | 5 +---- 3 files changed, 3 insertions(+), 7 deletions(-) diff --git a/asset_backtester/__init__.py b/asset_backtester/__init__.py index 216e997..0bd50a9 100644 --- a/asset_backtester/__init__.py +++ b/asset_backtester/__init__.py @@ -1,2 +1 @@ -from backtester import Backtest -from signal import Signal, get_order \ No newline at end of file +from .backtester import Backtest diff --git a/asset_backtester/backtester.py b/asset_backtester/backtester.py index eb3ac4a..5eba9da 100644 --- a/asset_backtester/backtester.py +++ b/asset_backtester/backtester.py @@ -1,7 +1,7 @@ import pandas as pd import pyprind -from portfolio.portfolio import Portfolio +from portfolio import Portfolio class Backtest: diff --git a/asset_backtester/portfolio/portfolio.py b/asset_backtester/portfolio/portfolio.py index 0fc2032..eaddefd 100644 --- a/asset_backtester/portfolio/portfolio.py +++ b/asset_backtester/portfolio/portfolio.py @@ -1,11 +1,8 @@ -from .direction import Direction from .asset import Asset class Portfolio: - def __init__(self, direction=Direction.BUY): - assert isinstance(direction, Direction) - self.direction = direction + def __init__(self): self.assets = [] def add_asset(self, asset):