desiree commited on
Commit
a157de3
·
verified ·
1 Parent(s): 897318a

Fix lanes init: DEFAULT_REGISTRIES maps kind -> single renderer

Browse files
Files changed (1) hide show
  1. spi/renderers/registry.py +2 -1
spi/renderers/registry.py CHANGED
@@ -32,7 +32,8 @@ class ModelRouter:
32
  raise ValueError(f"unknown SPI_MODE '{mode}'; expected mock|small|production")
33
  self.mode = mode
34
  self._lanes: dict[str, dict[str, Renderer]] = {
35
- kind: dict(lanes) for kind, lanes in DEFAULT_REGISTRIES[mode].items()
 
36
  }
37
 
38
  def register(self, kind: str, renderer: Renderer) -> None:
 
32
  raise ValueError(f"unknown SPI_MODE '{mode}'; expected mock|small|production")
33
  self.mode = mode
34
  self._lanes: dict[str, dict[str, Renderer]] = {
35
+ kind: {renderer.name: renderer}
36
+ for kind, renderer in DEFAULT_REGISTRIES[mode].items()
37
  }
38
 
39
  def register(self, kind: str, renderer: Renderer) -> None: