From 23a9cf2cba8fc4db95de964564e22c1e9fae17eb Mon Sep 17 00:00:00 2001
From: JohnJim0816 <johnjim0816@gmail.com>
Date: Sat, 27 Feb 2021 16:58:40 +0800
Subject: [PATCH] fix Q-leaning

---
 codes/Q-learning/main.py | 5 -----
 1 file changed, 5 deletions(-)

diff --git a/codes/Q-learning/main.py b/codes/Q-learning/main.py
index 8d35b08..75618d7 100644
--- a/codes/Q-learning/main.py
+++ b/codes/Q-learning/main.py
@@ -136,11 +136,6 @@ def eval(cfg, saved_model_path = SAVED_MODEL_PATH):
     plot(rewards)
     plot(MA_rewards,ylabel='moving_average_rewards_train')
       
-def main():
-    cfg = get_args()
-    # train(cfg)
-    test(cfg)
-
 if __name__ == "__main__":
     cfg = get_args()
     if cfg.train: