--- a/rust/hg-pyo3/src/dirstate.rs	Wed Jan 29 14:03:24 2025 +0100
+++ b/rust/hg-pyo3/src/dirstate.rs	Wed Jan 29 14:05:21 2025 +0100
@@ -12,6 +12,8 @@
 //! From Python, this will be seen as `mercurial.pyo3_rustext.dirstate`
 use crate::{exceptions, utils::new_submodule};
 use pyo3::prelude::*;
+mod dirstate_map;
+use dirstate_map::DirstateIdentity;
 
 pub fn init_module<'py>(
     py: Python<'py>,
@@ -20,5 +22,6 @@
     let m = new_submodule(py, package, "dirstate")?;
     m.add("__doc__", "Dirstate - Rust implementation exposed via PyO3")?;
     m.add("FallbackError", py.get_type::<exceptions::FallbackError>())?;
+    m.add_class::<DirstateIdentity>()?;
     Ok(m)
 }