BaseRegistry

class tfsnippet.utils.BaseRegistry(ignore_case=False)

Bases: object

A base class for implement a type or object registry.

Usage:

registry = BaseRegistry()
registry.register('MNIST', spt.datasets.MNIST())

Attributes Summary

ignore_case Whether or not to ignore the case?

Methods Summary

get(name) Get an object.
register(name, obj) Register an object.

Attributes Documentation

ignore_case

Whether or not to ignore the case?

Methods Documentation

get(name)

Get an object.

Parameters:name (str) – Name of the object.
Returns:The retrieved object.
Raises:KeyError – If name is not registered.
register(name, obj)

Register an object.

Parameters:
  • name (str) – Name of the object.
  • obj – The object.