Fixed Bad Import Issue

notadamking · Jul 9, 2019 · be6d144 · be6d144
1 parent b2ed19a
commit be6d144
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 3 deletions.
diff --git a/lib/RLTrader.py b/lib/RLTrader.py
@@ -12,7 +12,7 @@
 from stable_baselines import PPO2
 
 from lib.env.TradingEnv import TradingEnv
-from lib.env.reward import BaseRewardStrategy, IncrementalProfit
+from lib.env.reward import BaseRewardStrategy, IncrementalProfit, WeightedUnrealisedProfit
 from lib.data.providers.dates import ProviderDateFormat
 from lib.data.providers import BaseDataProvider,  StaticDataProvider, ExchangeDataProvider
 from lib.util.logger import init_logger

diff --git a/lib/env/TradingEnv.py b/lib/env/TradingEnv.py
@@ -26,7 +26,7 @@ class TradingEnv(gym.Env):
 
     def __init__(self,
                  data_provider: BaseDataProvider,
-                 reward_strategy: BaseRewardStrategy = WeightedUnrealisedProfit,
+                 reward_strategy: BaseRewardStrategy = IncrementalProfit,
                  initial_balance: int = 10000,
                  commission: float = 0.0025,
                  **kwargs):

diff --git a/lib/env/reward/IncrementalProfit.py b/lib/env/reward/IncrementalProfit.py
@@ -7,7 +7,7 @@
 
 class IncrementalProfit(BaseRewardStrategy):
     def __init__(self):
-        pass
+        super(IncrementalProfit, self).__init__()
 
     def reset_reward(self):
         pass

diff --git a/lib/env/reward/__init__.py b/lib/env/reward/__init__.py
@@ -1,2 +1,3 @@
 from lib.env.reward.IncrementalProfit import IncrementalProfit
+from lib.env.reward.WeightedUnrealisedProfit import WeightedUnrealisedProfit
 from lib.env.reward.BaseRewardStrategy import BaseRewardStrategy