mirror of https://github.com/hpcaitech/ColossalAI
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
22 lines
786 B
22 lines
786 B
class SingletonMeta(type):
|
|
"""
|
|
The Singleton class can be implemented in different ways in Python. Some
|
|
possible methods include: base class, decorator, metaclass. We will use the
|
|
metaclass because it is best suited for this purpose.
|
|
"""
|
|
|
|
_instances = {}
|
|
|
|
def __call__(cls, *args, **kwargs):
|
|
"""
|
|
Possible changes to the value of the `__init__` argument do not affect
|
|
the returned instance.
|
|
"""
|
|
if cls not in cls._instances:
|
|
instance = super().__call__(*args, **kwargs)
|
|
cls._instances[cls] = instance
|
|
else:
|
|
assert len(args) == 0 and len(
|
|
kwargs) == 0, f'{cls.__name__} is a singleton class and a instance has been created.'
|
|
return cls._instances[cls]
|