Ethanfel/Comfyui-Nodes-Stats
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: atomic ModelMapper build and isolate test NODE_CLASS_MAPPINGS mutations

Browse Source 
Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
Ethanfel
2026-04-09 18:03:16 +02:00
parent 9891b3ec41
commit 38e95c150a
2 changed files with 14 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mapper.py
+10 -6
View File
@@ -92,7 +92,7 @@ class ModelMapper:
try:
import folder_paths
self._folder_files = {}
folder_files = {}
for folder_type in folder_paths.folder_names_and_paths:
if folder_type in EXCLUDED_FOLDER_TYPES:
continue
@@ -101,13 +101,17 @@ class ModelMapper:
except Exception:
files = []
if files:
self._folder_files[folder_type] = frozenset(files)
folder_files[folder_type] = frozenset(files)
# Reverse map: filename -> folder_type (last write wins on collision)
self._reverse = {}
for folder_type, files in self._folder_files.items():
# Build reverse map: filename -> folder_type (last write wins on collision)
reverse = {}
for folder_type, files in folder_files.items():
for f in files:
self._reverse[f] = folder_type
reverse[f] = folder_type
# Assign atomically only on success
self._folder_files = folder_files
self._reverse = reverse
except Exception:
logger.warning("ModelMapper: failed to build model map", exc_info=True)