--- a/rust/hg-pyo3/src/lib.rs	Wed Feb 12 09:37:55 2025 -0500
+++ b/rust/hg-pyo3/src/lib.rs	Fri Jan 24 15:58:18 2025 +0100
@@ -2,6 +2,7 @@
 
 mod ancestors;
 mod dagops;
+mod dirstate;
 mod discovery;
 mod exceptions;
 mod node;
@@ -23,6 +24,7 @@
 
     m.add_submodule(&ancestors::init_module(py, &dotted_name)?)?;
     m.add_submodule(&dagops::init_module(py, &dotted_name)?)?;
+    m.add_submodule(&dirstate::init_module(py, &dotted_name)?)?;
     m.add_submodule(&discovery::init_module(py, &dotted_name)?)?;
     m.add_submodule(&revlog::init_module(py, &dotted_name)?)?;
     m.add("GraphError", py.get_type::<exceptions::GraphError>())?;