Skip to content

Commit b6242d9

Browse files
committed
fix: remove extra parameter in accelerator registry decorator
1 parent c6b6553 commit b6242d9

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/lightning/fabric/accelerators/registry.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -73,14 +73,14 @@ def register(
7373
data["description"] = description
7474
data["init_params"] = init_params
7575

76-
def do_register(name: str, accelerator: Callable) -> Callable:
76+
def do_register(accelerator: Callable) -> Callable:
7777
data["accelerator"] = accelerator
7878
data["accelerator_name"] = name
7979
self[name] = data
8080
return accelerator
8181

8282
if accelerator is not None:
83-
return do_register(name, accelerator)
83+
return do_register(accelerator)
8484

8585
return do_register
8686

0 commit comments

Comments
 (0)