art-dag/celery
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Don't overwrite pre-registered primitives in _load_primitives

Browse Source
This commit is contained in:
giles
2026-02-02 23:59:38 +00:00
parent d8360e5945
commit 3d5a08a7dc
1 changed files with 9 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
streaming/stream_sexp_generic.py
View File

@@ -189,7 +189,10 @@ class StreamInterpreter:
prim_name = name[5:] prim_name = name[5:]
dash_name = prim_name.replace('_', '-') dash_name = prim_name.replace('_', '-')
# Register ONLY with namespace (geometry:ripple-displace) # Register ONLY with namespace (geometry:ripple-displace)
self.primitives[f"{lib_name}:{dash_name}"] = func # Don't overwrite if already registered (allows pre-registration of overrides)
key = f"{lib_name}:{dash_name}"
if key not in self.primitives:
self.primitives[key] = func
count += 1 count += 1
if hasattr(module, 'PRIMITIVES'): if hasattr(module, 'PRIMITIVES'):
@@ -197,9 +200,12 @@ class StreamInterpreter:
if isinstance(prims, dict): if isinstance(prims, dict):
for name, func in prims.items(): for name, func in prims.items():
# Register ONLY with namespace # Register ONLY with namespace
# Don't overwrite if already registered
dash_name = name.replace('_', '-') dash_name = name.replace('_', '-')
self.primitives[f"{lib_name}:{dash_name}"] = func key = f"{lib_name}:{dash_name}"
count += 1 if key not in self.primitives:
self.primitives[key] = func
count += 1
print(f"Loaded primitives: {lib_name} ({count} functions)", file=sys.stderr) print(f"Loaded primitives: {lib_name} ({count} functions)", file=sys.stderr)