Use strtobool

LucasAlegre · Feb 15, 2024 · d21f19d · d21f19d
1 parent 815425c
commit d21f19d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/morl_baselines/common/morl_algorithm.py b/morl_baselines/common/morl_algorithm.py
@@ -2,6 +2,7 @@
 import os
 import time
 from abc import ABC, abstractmethod
+from distutils.util import strtobool
 from typing import Dict, Optional, Union
 
 import gymnasium as gym
@@ -261,7 +262,7 @@ def setup_wandb(
         config = self.get_config()
         config["algo"] = self.experiment_name
         # looks for whether we're using a Gymnasium based env in env_variable
-        monitor_gym = os.environ.get("MONITOR_GYM", True)
+        monitor_gym = strtobool(os.environ.get("MONITOR_GYM", "True"))
 
         wandb.init(
             project=project_name,