Merge pull request #94 from LucasAlegre/chore/remove_tensorboard_stuf…

…f_make_gym_logging_optional

Chore/remove tensorboard stuff make gym logging optional

morl_baselines/common/morl_algorithm.py

@@ -1,6 +1,8 @@
 """MORL algorithm base classes."""
+import os
 import time
 from abc import ABC, abstractmethod
+from distutils.util import strtobool
 from typing import Dict, Optional, Union
 
 import gymnasium as gym
@@ -259,14 +261,15 @@ 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 = strtobool(os.environ.get("MONITOR_GYM", "True"))
 
         wandb.init(
             project=project_name,
             entity=entity,
-            sync_tensorboard=True,
             config=config,
             name=self.full_experiment_name,
-            monitor_gym=True,
+            monitor_gym=monitor_gym,
             save_code=True,
             group=group,
         )

morl_baselines/multi_policy/linear_support/linear_support.py

@@ -385,13 +385,11 @@ def _solve(w):
 
     num_objectives = 2
     ols = LinearSupport(num_objectives=num_objectives, epsilon=0.0001, verbose=True)
-    while True:
-        w = ols.next_weight()
-        if ols.ended():
-            print("OLS ended.")
-            break
+    w = ols.next_weight()
+    while not ols.ended():
         print("w:", w)
         value = _solve(w)
         ols.add_solution(value, w)
 
         print("hv:", hypervolume(np.zeros(num_objectives), ols.ccs))
+        w = ols.next_weight()