Fix import

deep_rl/network/network_utils.py

@@ -7,6 +7,7 @@
 import torch
 import torch.nn as nn
 import torch.nn.functional as F
+import numpy as np
 
 class BaseNet:
     def set_gpu(self, gpu):

deep_rl/utils/plot.py

@@ -1,7 +1,7 @@
 # Adapted from  https://github.com/openai/baselines/blob/master/baselines/results_plotter.py
 
+from ..component.bench import load_monitor_log
 import numpy as np
-from ..component import *
 import os
 import re
 
@@ -44,7 +44,7 @@ class Plotter:
     def load_results(self, dirs, max_timesteps=1e8, x_axis=X_TIMESTEPS, episode_window=100):
         tslist = []
         for dir in dirs:
-            ts = component.load_monitor_log(dir)
+            ts = load_monitor_log(dir)
             ts = ts[ts.l.cumsum() <= max_timesteps]
             tslist.append(ts)
         xy_list = [self.ts2xy(ts, x_axis) for ts in tslist]