Fixed issue 840 (#841)

flow-project · Mar 19, 2020 · 4c49ab7 · 4c49ab7
1 parent d1688c6
commit 4c49ab7
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/flow/utils/rllib.py b/flow/utils/rllib.py
@@ -6,6 +6,7 @@
 import json
 from copy import deepcopy
 import os
+import sys
 
 import flow.envs
 from flow.core.params import SumoLaneChangeParams, SumoCarFollowingParams, \
@@ -207,6 +208,9 @@ def get_rllib_config(path):
 
 def get_rllib_pkl(path):
     """Return the data from the specified rllib configuration file."""
+    dirname = os.path.dirname(__file__)
+    filename = os.path.join(dirname, '../../examples/')
+    sys.path.append(filename)
     config_path = os.path.join(path, "params.pkl")
     if not os.path.exists(config_path):
         config_path = os.path.join(path, "../params.pkl")