diff --git a/tools/cutover/carry_finished_turns.py b/tools/cutover/carry_finished_turns.py index c6ff817133..2c3f91c8fd 100644 --- a/tools/cutover/carry_finished_turns.py +++ b/tools/cutover/carry_finished_turns.py @@ -14,7 +14,7 @@ from agent_comms.historical_views import HistoryArchive from agent_comms.installer.carry import Carried, Carry from agent_comms.registry_store import RegistryStore -from agent_comms.store_files import _atomic_write_text, _store_lock +from agent_comms.store_files import AtomicFile, _store_lock from .carry_bus_history import BusHistory, CarryCatalogSource from .carry_dead_turns import DeadTurns @@ -63,8 +63,8 @@ def carry(self, *, apply: bool) -> Carried: for name in result.rewritten ] if resealed: - _atomic_write_text( - HistoryArchive.path_in(self.root), json.dumps(FieldCodec.encode(tuple(rebound))) + AtomicFile(HistoryArchive.path_in(self.root)).write_text( + json.dumps(FieldCodec.encode(tuple(rebound))) ) rewritten.append(HistoryArchive.filename) finished = sum(int(result.detail.get("finished", 0)) for _, result in results)