From d0a9f9900dfcaffcd81fcbb2bf0ac902ceb99198 Mon Sep 17 00:00:00 2001
From: Mike Clark <wassname@users.noreply.github.com>
Date: Sun, 29 Oct 2017 13:29:47 +0800
Subject: [PATCH] register env's

---
 src/environments/__init__.py | 31 +++++++++++++++++++++++++++++++
 1 file changed, 31 insertions(+)

diff --git a/src/environments/__init__.py b/src/environments/__init__.py
index e69de29..d3d3e35 100644
--- a/src/environments/__init__.py
+++ b/src/environments/__init__.py
@@ -0,0 +1,31 @@
+import pandas as pd
+from gym.envs.registration import register
+
+df_train = pd.read_hdf('./data/poloniex_30m.hf', key='train')
+
+register(
+    id='PortfolioMLP-v0',
+    entry_point='src.environments.portfolio:PortfolioEnv',
+    kwargs=dict(
+        output_mode='mlp',
+        df=df_train
+    )
+)
+
+register(
+    id='PortfolioEIIE-v0',
+    entry_point='src.environments.portfolio:PortfolioEnv',
+    kwargs=dict(
+        output_mode='EIIE',
+        df=df_train
+    )
+)
+
+register(
+    id='PortfolioAtari-v0',
+    entry_point='src.environments.portfolio:PortfolioEnv',
+    kwargs=dict(
+        output_mode='atari',
+        df=df_train
+    )
+)