fix dt checking

LeCAR-Lab · Feb 12, 2025 · 83b3c61 · 83b3c61
1 parent f770410
commit 83b3c61
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dial_mpc/envs/base_env.py b/dial_mpc/envs/base_env.py
@@ -13,7 +13,7 @@
 
 class BaseEnv(PipelineEnv):
     def __init__(self, config: BaseEnvConfig):
-        assert config.dt % config.timestep == 0, "timestep must be divisible by dt"
+        assert jnp.allclose(config.dt % config.timestep, 0.0), "timestep must be divisible by dt"
         self._config = config
         n_frames = int(config.dt / config.timestep)
         sys = self.make_system(config)