Support python3 and pytorch 0.2

component/__init__.py

@@ -1,5 +1,5 @@
-from atari_wrapper import *
-from policy import *
-from replay import *
-from task import *
-from random_process import *
+from .atari_wrapper import *
+from .policy import *
+from .replay import *
+from .task import *
+from .random_process import *

component/task.py

@@ -6,7 +6,7 @@
 import gym
 import sys
 import numpy as np
-from atari_wrapper import *
+from .atari_wrapper import *
 
 class BasicTask:
     def __init__(self):