Rasa.core.agent import Agent Issue

I’m using google Colab. It appears after this code:

from rasa.core.interpreter import RasaNLUInterpreter 
from rasa.core.policies.policy import *
from rasa.core.agent import Agent
from rasa.core import training

agent = Agent('domain.yml', policies=[MemoizationPolicy(),  MappingPolicy(), TEDPolicy()])

# Configuration for Rasa Core.

# https://rasa.com/docs/rasa/core/policies/

# loading our neatly defined training dialogues

training_data = agent.load_data('stories.md')

agent.train(

    training_data,

    validation_split=0.2,

    epochs=200, 

    max_history= 5
)

agent.persist('models/dialogue')