From 835a2e88002492ece703af7d4d04e65143ee816b Mon Sep 17 00:00:00 2001
From: Ivan-267 <61947090+Ivan-267@users.noreply.github.com>
Date: Fri, 1 Dec 2023 17:34:46 +0100
Subject: [PATCH] Fix conditional statement

---
 examples/stable_baselines3_example.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/examples/stable_baselines3_example.py b/examples/stable_baselines3_example.py
index dbea0099..cf208b77 100644
--- a/examples/stable_baselines3_example.py
+++ b/examples/stable_baselines3_example.py
@@ -151,7 +151,7 @@ def func(progress_remaining: float) -> float:
 
 
 if args.resume_model_path is None:
-    learning_rate = 0.0003 if args.inference else linear_schedule(0.0003)
+    learning_rate = 0.0003 if not args.linear_lr_schedule else linear_schedule(0.0003)
     model: PPO = PPO("MultiInputPolicy",
                      env,
                      ent_coef=0.0001,