From afc3719ab94527ccb956ade1439ae43cbd2b7853 Mon Sep 17 00:00:00 2001
From: Johannes Fischer <johannes.fischer@kit.edu>
Date: Thu, 29 Jul 2021 15:30:55 +0200
Subject: [PATCH] Fix bug with wrong argument order

---
 src/main.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/main.py b/src/main.py
index 76e83c7..c924d28 100644
--- a/src/main.py
+++ b/src/main.py
@@ -54,7 +54,7 @@ def main(config, method='bc', train=False, test=False, loc=0, datadir='./expert_
     if test:
 
         # load policy 
-        policy = policy_class.load_model(config, filestr)
+        policy = policy_class.load_model(filestr, config)
         policy.eval()
 
         # simulate policy