add missing Python classes in feos.ideal_gas

feos-org · Nov 28, 2023 · 612f97f · 612f97f
1 parent d0a8c82
commit 612f97f
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/python/ideal_gas.rs b/src/python/ideal_gas.rs
@@ -1,8 +1,14 @@
-use feos_core::python::joback::{PyJobackParameters, PyJobackRecord};
+use feos_core::parameter::IdentifierOption;
+use feos_core::python::joback::*;
+use feos_core::python::parameter::PyIdentifier;
 use pyo3::prelude::*;
 
 #[pymodule]
 pub fn ideal_gas(_py: Python<'_>, m: &PyModule) -> PyResult<()> {
+    m.add_class::<PyIdentifier>()?;
+    m.add_class::<IdentifierOption>()?;
     m.add_class::<PyJobackRecord>()?;
+    m.add_class::<PyPureRecord>()?;
+    m.add_class::<PySegmentRecord>()?;
     m.add_class::<PyJobackParameters>()
 }