Add fire to test_simple_rl.py (#4378)

Co-authored-by: Vincent-Pierre BERGES <[email protected]>
Co-authored-by: Ervin T <[email protected]>

Author: 3 people

ml-agents/mlagents/trainers/policy/policy.py

@@ -43,11 +43,6 @@ def __init__(
         self.vis_obs_size = sum(
             1 for shape in behavior_spec.observation_shapes if len(shape) == 3
         )
-        self.vis_obs_shape = (
-            [shape for shape in behavior_spec.observation_shapes if len(shape) == 3][0]
-            if self.vis_obs_size > 0
-            else None
-        )
         self.use_continuous_act = behavior_spec.is_action_continuous()
         self.num_branches = self.behavior_spec.action_size
         self.previous_action_dict: Dict[str, np.array] = {}

ml-agents/mlagents/trainers/tests/test_simple_rl.py

@@ -27,6 +27,7 @@
     RewardSignalType,
     EncoderType,
     ScheduleType,
+    FrameworkType,
 )
 from mlagents.trainers.environment_parameter_manager import EnvironmentParameterManager
 from mlagents_envs.side_channel.environment_parameters_channel import (
@@ -53,6 +54,7 @@
     summary_freq=500,
     max_steps=3000,
     threaded=False,
+    framework=FrameworkType.TENSORFLOW,
 )
 
 SAC_CONFIG = TrainerSettings(