From 612f97fbc5ebe10bfc7145596907913095b52af4 Mon Sep 17 00:00:00 2001
From: Philipp Rehner <prehner@ethz.ch>
Date: Tue, 28 Nov 2023 16:17:11 +0100
Subject: [PATCH] add missing Python classes in feos.ideal_gas

---
 src/python/ideal_gas.rs | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/src/python/ideal_gas.rs b/src/python/ideal_gas.rs
index 04e0d177c..bbcc8aefc 100644
--- 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>()
 }