fix in utils for getting sim

lasgroup · Nov 30, 2023 · 6b8c721 · 6b8c721
1 parent e6321aa
commit 6b8c721
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/experiments/online_rl_hardware/utils.py b/experiments/online_rl_hardware/utils.py
@@ -136,7 +136,8 @@ def set_up_model_based_sac_trainer(bnn_model, data_buffer, data_buffer_state, ke
 
 
 def set_up_bnn_dynamics_model(config: Any, key: jax.random.PRNGKey):
-    sim = RaceCarSim(encode_angle=True, use_blend=config.sim_prior == 'high_fidelity', car_id=2)
+    use_blend = 'high_fidelity' in config.sim_prior
+    sim = RaceCarSim(encode_angle=True, use_blend=use_blend, car_id=2)
     if config.num_stacked_actions > 0:
         sim = StackedActionSimWrapper(sim, num_stacked_actions=config.num_stacked_actions, action_size=2)
     if config.predict_difference: