//! This program is used to quickly check pybullet works properly with pyo3.

use anyhow::Result;
use pyo3::{types::IntoPyDict, Python};
// With a version of tch, commenting out the following line causes segmentation fault.
// use tch::Tensor;

fn main() -> Result<()> {
    Python::with_gil(|py| {
        let gym = py.import("gym")?;
        py.import("pybulletgym")?;
        let env = gym.call_method("make", ("AntPyBulletEnv-v0",), None)?;
        let kwargs = vec![("mode", "human")].into_py_dict(py);
        env.call_method("render", (), Some(kwargs))?;

        env.call_method0("reset")?;
        loop {}

        #[allow(unreachable_code)]
        Ok(())
    })
}