Diagram import error

[vijaybalamahalingam]

For the below code "from lambeq import Diagram" isn't working

simple_diagram = Diagram.tensor(
[AtomicType.NOUN, AtomicType.NOUN],
[AtomicType.SENTENCE]
)

[dimkart]

Try

from lambeq.backend.grammar import Diagram

Also, the tensor method accepts diagrams.