mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-09-03 10:06:44 +00:00
[misc] update pre-commit and run all files (#4752)
* [misc] update pre-commit * [misc] run pre-commit * [misc] remove useless configuration files * [misc] ignore cuda for clang-format
This commit is contained in:
@@ -1,14 +1,12 @@
|
||||
__all__ = ['Registry']
|
||||
__all__ = ["Registry"]
|
||||
|
||||
|
||||
class Registry:
|
||||
|
||||
def __init__(self, name):
|
||||
self.name = name
|
||||
self.store = {}
|
||||
|
||||
def register(self, source):
|
||||
|
||||
def wrapper(func):
|
||||
if isinstance(source, (list, tuple)):
|
||||
# support register a list of items for this func
|
||||
@@ -21,7 +19,7 @@ class Registry:
|
||||
return wrapper
|
||||
|
||||
def get(self, source):
|
||||
assert source in self.store, f'{source} not found in the {self.name} registry'
|
||||
assert source in self.store, f"{source} not found in the {self.name} registry"
|
||||
target = self.store[source]
|
||||
return target
|
||||
|
||||
@@ -29,4 +27,4 @@ class Registry:
|
||||
return source in self.store
|
||||
|
||||
|
||||
meta_register = Registry('meta')
|
||||
meta_register = Registry("meta")
|
||||
|
Reference in New Issue
Block a user