amend

pytorch · vmoens · Jan 31, 2024 · Jan 19, 2024 · Jan 31, 2024 · Jan 31, 2024
commit f68dd4faa34f8396f94041ff25c594bc2482b957
diff --git a/.github/unittest/linux_examples/scripts/run_test.sh b/.github/unittest/linux_examples/scripts/run_test.sh
@@ -319,7 +319,7 @@ python .github/unittest/helpers/coverage_run_parallel.py examples/multiagent/iql
   collector.frames_per_batch=200 \
   train.num_epochs=3 \
   train.minibatch_size=100 \
-  logger.backend=
+  logger.backendg
 python .github/unittest/helpers/coverage_run_parallel.py examples/multiagent/qmix_vdn.py \
   collector.n_iters=2 \
   collector.frames_per_batch=200 \

diff --git a/examples/td3/td3.py b/examples/td3/td3.py
@@ -30,7 +30,7 @@
     make_replay_buffer,
     make_td3_agent,
 )
-
+from torchrl._utils import logger as torchrl_logger
 
 @hydra.main(version_base="1.1", config_path=".", config_name="config")
 def main(cfg: "DictConfig"):  # noqa: F821
@@ -206,7 +206,7 @@ def main(cfg: "DictConfig"):  # noqa: F821
     collector.shutdown()
     end_time = time.time()
     execution_time = end_time - start_time
-    logger.info(f"Training took {execution_time:.2f} seconds to finish")
+    torchrl_logger.info(f"Training took {execution_time:.2f} seconds to finish")
 
 
 if __name__ == "__main__":