--- a/rust/hg-pyo3/src/dirstate.rs	Thu Jan 30 12:24:41 2025 +0100
+++ b/rust/hg-pyo3/src/dirstate.rs	Tue Feb 04 10:42:26 2025 +0100
@@ -19,6 +19,8 @@
     DirstateIdentity, DirstateMap, DirstateMapItemsIterator,
     DirstateMapKeysIterator,
 };
+mod copy_map;
+use copy_map::CopyMap;
 
 pub fn init_module<'py>(
     py: Python<'py>,
@@ -32,5 +34,6 @@
     m.add_class::<DirstateMap>()?;
     m.add_class::<DirstateMapKeysIterator>()?;
     m.add_class::<DirstateMapItemsIterator>()?;
+    m.add_class::<CopyMap>()?;
     Ok(m)
 }