tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

improve handling of file lock when content is empty (#2959)

Browse Source
This commit is contained in:
Wing Lian
2025-07-24 16:10:38 -04:00
committed by GitHub
parent e80faea0db
commit 460e0f9ed9
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/axolotl/utils/data/lock.py
View File

@@ -46,7 +46,8 @@ class FileLockLoader:
def _increment_counter(self): def _increment_counter(self):
"""Safely increment the process counter.""" """Safely increment the process counter."""
if self.counter_path.exists(): if self.counter_path.exists():
count = int(self.counter_path.read_text().strip()) counter_content = self.counter_path.read_text().strip()
count = int(counter_content) if counter_content else 0
else: else:
count = 0 count = 0
self.counter_path.write_text(str(count + 1)) self.counter_path.write_text(str(count + 1))
@@ -54,10 +55,11 @@ class FileLockLoader:
def cleanup(self): def cleanup(self):
"""Clean up ready flag when last process is done.""" """Clean up ready flag when last process is done."""
with FileLock(str(self.lock_file_path)): with FileLock(str(self.lock_file_path)):
count = int(self.counter_path.read_text().strip()) counter_content = self.counter_path.read_text().strip()
count = int(counter_content) if counter_content else 0
count -= 1 count -= 1
if count == 0: if count <= 0:
# Last process cleans everything up # Last process cleans everything up
self.ready_flag_path.unlink(missing_ok=True) self.ready_flag_path.unlink(missing_ok=True)
self.counter_path.unlink(missing_ok=True) self.counter_path.unlink(missing_ok=True)