--- a/rust/hg-cpython/src/dirstate.rs	Mon Oct 14 14:14:21 2024 +0200
+++ b/rust/hg-cpython/src/dirstate.rs	Thu Oct 03 16:35:31 2024 +0200
@@ -16,12 +16,11 @@
 mod status;
 use self::item::DirstateItem;
 use crate::{
-    dirstate::{
-        dirs_multiset::Dirs, dirstate_map::DirstateMap, status::status_wrapper,
-    },
+    dirstate::{dirs_multiset::Dirs, status::status_wrapper},
     exceptions,
 };
 use cpython::{PyBytes, PyDict, PyList, PyModule, PyObject, PyResult, Python};
+use dirstate_map::{DirstateIdentity, DirstateMap};
 use hg::dirstate_tree::on_disk::V2_FORMAT_MARKER;
 
 /// Create the module, with `__package__` given from parent
@@ -42,6 +41,7 @@
     m.add_class::<Dirs>(py)?;
     m.add_class::<DirstateMap>(py)?;
     m.add_class::<DirstateItem>(py)?;
+    m.add_class::<DirstateIdentity>(py)?;
     m.add(py, "V2_FORMAT_MARKER", PyBytes::new(py, V2_FORMAT_MARKER))?;
     m.add(
         py,