class AttentionBank:
    def __init__(self, save_steps, block_map, inject_steps=None):
        self._data = {
            "save_steps": save_steps,
            "block_map": block_map,
            "inject_steps": inject_steps,
        }

    def __getitem__(self, key):
        return self._data[key]

    def __setitem__(self, key, value):
        self._data[key] = value

    def get(self, key, default=None):
        return self._data.get(key, default)