Fix BaseRewardStrategy inheritance.

notadamking · Jul 6, 2019 · 2422f03 · 2422f03
1 parent f06dd8f
commit 2422f03
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/lib/env/reward/BaseRewardStrategy.py b/lib/env/reward/BaseRewardStrategy.py
@@ -5,8 +5,8 @@
 
 
 class BaseRewardStrategy(object, metaclass=ABCMeta):
-    @abstractmethod
     @staticmethod
+    @abstractmethod
     def get_reward(observations: pd.DataFrame,
                    account_history: pd.DataFrame,
                    net_worths: List[float],

diff --git a/lib/env/reward/IncrementalProfit.py b/lib/env/reward/IncrementalProfit.py
@@ -2,7 +2,7 @@
 
 from typing import List
 
-from lib.env.reward import BaseRewardStrategy
+from lib.env.reward.BaseRewardStrategy import BaseRewardStrategy
 
 
 class IncrementalProfit(BaseRewardStrategy):