We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent c6b6553 commit b6242d9Copy full SHA for b6242d9
src/lightning/fabric/accelerators/registry.py
@@ -73,14 +73,14 @@ def register(
73
data["description"] = description
74
data["init_params"] = init_params
75
76
- def do_register(name: str, accelerator: Callable) -> Callable:
+ def do_register(accelerator: Callable) -> Callable:
77
data["accelerator"] = accelerator
78
data["accelerator_name"] = name
79
self[name] = data
80
return accelerator
81
82
if accelerator is not None:
83
- return do_register(name, accelerator)
+ return do_register(accelerator)
84
85
return do_register
86
0 commit comments