@@ -907,13 +907,16 @@ def test_reproduce_multiple_episodes(self):
907
907
# Check that the arguments for the new experiment are correct
908
908
###
909
909
910
+ # NOTE: detailed wandb logging is currently removed as default. If the handler
911
+ # is added back, the handler should be removed in unit tests again. (also in the
912
+ # test_reproduce_single_episode_with_multiple_episode_function)
910
913
# Detailed wandb logging should be automatically built in, though we will remove
911
914
# it to avoid logging tests to wandb
912
- self .assertEqual (
913
- eval_cfg_2 ["logging_config" ]["wandb_handlers" ][- 1 ],
914
- DetailedWandbMarkedObsHandler ,
915
- )
916
- eval_cfg_2 ["logging_config" ]["wandb_handlers" ].pop ()
915
+ # self.assertEqual(
916
+ # eval_cfg_2["logging_config"]["wandb_handlers"][-1],
917
+ # DetailedWandbMarkedObsHandler,
918
+ # )
919
+ # eval_cfg_2["logging_config"]["wandb_handlers"].pop()
917
920
918
921
# capsule3DSolid is used as the lone eval object; make sure it is listed once
919
922
# per episode
@@ -1023,14 +1026,6 @@ def test_reproduce_single_episode_with_multiple_episode_function(self):
1023
1026
# Check that the arguments for the new experiment are correct
1024
1027
###
1025
1028
1026
- # Detailed wandb logging should be automatically built in, though we will remove
1027
- # it to avoid logging tests to wandb
1028
- self .assertEqual (
1029
- eval_cfg_2 ["logging_config" ]["wandb_handlers" ][- 1 ],
1030
- DetailedWandbMarkedObsHandler ,
1031
- )
1032
- eval_cfg_2 ["logging_config" ]["wandb_handlers" ].pop ()
1033
-
1034
1029
# check that the object being used is the same one from original exp
1035
1030
self .assertEqual (
1036
1031
eval_cfg_1 ["eval_dataloader_args" ].object_names ,
0 commit comments