Allow Python subclassing of ComponentGraphConfig and ComponentGraph

shsms · shsms · commit f850440a03df · 2025-12-16T12:44:57.000+01:00
Signed-off-by: Sahas Subramanian &lt;sahas.subramanian@proton.me&gt;
diff --git a/src/graph.rs b/src/graph.rs
@@ -16,7 +16,7 @@ use pyo3::{
     types::{PyAny, PySet, PyType},
 };
 
-#[pyclass]
+#[pyclass(subclass)]
 #[derive(Clone, Default, Debug)]
 pub struct ComponentGraphConfig {
     config: cg::ComponentGraphConfig,
@@ -49,7 +49,7 @@ impl ComponentGraphConfig {
     }
 }
 
-#[pyclass]
+#[pyclass(subclass)]
 pub struct ComponentGraph {
     graph: cg::ComponentGraph<Component, Connection>,
 }

Original file line number	Diff line number	Diff line change
`@@ -16,7 +16,7 @@ use pyo3::{`
`16`	`16`	`types::{PyAny, PySet, PyType},`
`17`	`17`	`};`
`18`	`18`
`19`		`-#[pyclass]`
	`19`	`+#[pyclass(subclass)]`
`20`	`20`	`#[derive(Clone, Default, Debug)]`
`21`	`21`	`pub struct ComponentGraphConfig {`
`22`	`22`	`config: cg::ComponentGraphConfig,`
`@@ -49,7 +49,7 @@ impl ComponentGraphConfig {`
`49`	`49`	`}`
`50`	`50`	`}`
`51`	`51`
`52`		`-#[pyclass]`
	`52`	`+#[pyclass(subclass)]`
`53`	`53`	`pub struct ComponentGraph {`
`54`	`54`	`graph: cg::ComponentGraph<Component, Connection>,`
`55`	`55`	`}`