atomcloud.common package

Submodules

• atomcloud.common.registry module
  • Registry
    • Registry.get()
    • Registry.register()

Module contents

Created on Mon Mar 14 19:26:59 2022

@author: hofer

class atomcloud.common.Registry(name)

Bases: Iterable[Tuple[str, Any]]

The registry that provides name -> object mapping, to support third-party users’ custom modules.

To create a registry (e.g. a backbone registry):

BACKBONE_REGISTRY = Registry('BACKBONE')

To register an object:

@BACKBONE_REGISTRY.register()
class MyBackbone():
    ...

Or:

BACKBONE_REGISTRY.register(MyBackbone)

Parameters:

name (str) –

get(name)

Parameters:

name (str) –

Return type:

Any

register(name, obj=None)

Register the given object under the the name obj.__name__. Can be used as either a decorator or not. See docstring of this class for usage.

Parameters:

• name (str) –

• obj (Any | None) –

Return type:

Any