Skip to content

Commit 5628c29

Browse files
Merge branch 'master' of github.com:flow-project/flow into visuallizer_fix
2 parents 09b3585 + 4c49ab7 commit 5628c29

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

flow/utils/rllib.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77
import json
88
import sys
99
import os
10+
import sys
1011

1112
import flow.envs
1213
from flow.core.params import SumoLaneChangeParams, SumoCarFollowingParams, \
@@ -208,7 +209,7 @@ def get_rllib_config(path):
208209

209210
def get_rllib_pkl(path):
210211
"""Return the data from the specified rllib configuration file."""
211-
dirname = os.getcwd()
212+
dirname = os.path.dirname(__file__)
212213
filename = os.path.join(dirname, '../../examples/')
213214
sys.path.append(filename)
214215
config_path = os.path.join(path, "params.pkl")

0 commit comments

Comments
 (0)