Remove wblogger for now.

README.md

@@ -7,7 +7,7 @@
 [pypi-versions-badge]: https://img.shields.io/pypi/pyversions/dm_env_wrappers
 [pypi-badge]: https://badge.fury.io/py/dm_env_wrappers.svg
 [pypi]: https://pypi.org/project/dm_env_wrappers/
-[tests-badge]: https://img.shields.io/github/actions/workflow/status/kevinzakka/dm_env_wrappers/test.yml?branch=main
+[tests-badge]: https://github.com/kevinzakka/dm_env_wrappers/actions/workflows/build.yml/badge.svg
 [tests]: https://github.com/kevinzakka/dm_env_wrappers/actions/workflows/build.yml
 
 `dm_env_wrappers` is a collection of frequently-used wrappers for environments that respect the [dm_env](https://github.com/deepmind/dm_env) interface (e.g., dm_control suite/composer envs).

dm_env_wrappers/__init__.py

@@ -17,7 +17,6 @@
 from dm_env_wrappers._src.step_limit import StepLimitWrapper
 from dm_env_wrappers._src.validate_spec import ValidateActionSpecWrapper
 from dm_env_wrappers._src.video import VideoWrapper
-from dm_env_wrappers._src.wandb_logger import WandbLoggerWrapper, wblog
 
 __version__ = "0.0.6"
 
@@ -35,8 +34,6 @@
     "StepLimitWrapper",
     "ValidateActionSpecWrapper",
     "VideoWrapper",
-    "WandbLoggerWrapper",
-    "wblog",
     "wrap_all",
 )
 

dm_env_wrappers/_src/episode_statistics.py

@@ -5,7 +5,7 @@
 
 import dm_env
 
-from dm_env_wrappers._src import base, wandb_logger
+from dm_env_wrappers._src import base
 
 
 class EpisodeStatisticsWrapper(base.EnvironmentWrapper):
@@ -43,7 +43,6 @@ def step(self, action) -> dm_env.TimeStep:
             self._episode_length = 0
         return timestep
 
-    @wandb_logger.wblog
     def get_statistics(self) -> Dict[str, float]:
         """Returns the mean return and length of the last `deque_size` episodes."""
         if not self._return_queue or not self._length_queue: