<SYSTEM_TASK:>
Solve the following problem using Python, implementing the functions described below, one line at a time
<END_TASK>
<USER_TASK:>
Description:
 def get_state_generator(self, action_name, policy, kwargs):
 """ Returns the state generator to be used for the given action. :param action_name: Action identifier name. :type action_name: unicode | str :param policy: An instance of the current policy class. :type policy: dockermap.map.policy.base.BasePolicy :param kwargs: Keyword arguments. Can be modified by the initialization of the state generator. :type kwargs: dict :return: State generator object. :rtype: dockermap.map.state.base.AbstractStateGenerator """

        state_generator_cls = self.generators[action_name][0]
        state_generator = state_generator_cls(policy, kwargs)
        return state_generator