mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-09-22 01:48:07 +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,11 +1,9 @@
|
||||
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
|
||||
@@ -18,7 +16,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
|
||||
|
||||
@@ -26,4 +24,4 @@ class Registry:
|
||||
return source in self.store
|
||||
|
||||
|
||||
operator_registry = Registry('operator')
|
||||
operator_registry = Registry("operator")
|
||||
|
Reference in New Issue
Block a user