| import os | |
| def cleanup_dir(output_dir): | |
| # Walk through directory and remove all .lock files | |
| for root, dirs, files in os.walk(output_dir): | |
| for file in files: | |
| if file.endswith(".lock"): | |
| lock_file = os.path.join(root, file) | |
| try: | |
| os.remove(lock_file) | |
| except OSError: | |
| pass # Ignore errors if file can't be removed | |