def set_num_to_retrieve_to_contextualize_importance(
      self, num_to_retrieve: int) -> None:
    """Sets the number of memories to retrieve for contextualizing importance.

    Set this to 0 if you want to disable contextualization of importance.

    Args:
      num_to_retrieve: the number of memories to retrieve for contextualizing
        importance.
    """
    self._num_to_retrieve_to_contextualize_importance = num_to_retrieve
Sets the number of memories to retrieve for contextualizing importance.

    Set this to 0 if you want to disable contextualization of importance.

    Args:
      num_to_retrieve: the number of memories to retrieve for contextualizing
        importance.
    
concordia/associative_memory/deprecated/associative_memory.py