riiswa commited on
Commit
c11f67f
1 Parent(s): 72c067e

Try to fix type error

Browse files
Files changed (1) hide show
  1. app.py +0 -3
app.py CHANGED
@@ -50,9 +50,6 @@ if __name__ == "__main__":
50
  gr.Warning(
51
  "We're currently in the process of adding support for Mujoco environments, so the application may encounter crashes during this phase. We encourage contributors to join us in the repository https://github.com/riiswa/kanrl to assist in the development and support of other environments. Your contributions are invaluable in ensuring a robust and comprehensive framework."
52
  )
53
- torch.set_default_dtype(torch.float64)
54
- else:
55
- torch.set_default_dtype(torch.float32)
56
  agent = "ppo"
57
  if env_name == "Swimmer-v3" or env_name == "Walker2d-v3":
58
  agent ="trpo"
 
50
  gr.Warning(
51
  "We're currently in the process of adding support for Mujoco environments, so the application may encounter crashes during this phase. We encourage contributors to join us in the repository https://github.com/riiswa/kanrl to assist in the development and support of other environments. Your contributions are invaluable in ensuring a robust and comprehensive framework."
52
  )
 
 
 
53
  agent = "ppo"
54
  if env_name == "Swimmer-v3" or env_name == "Walker2d-v3":
55
  agent ="trpo"