from __future__ import annotations

from dataclasses import dataclass, field
from typing import Dict, Union

from .diagram_element import DiagramElement




@dataclass
class Diagram:
    """
    Container for a full IMR diagram: nodes and the links between them.
    """

    id: str
    nodes: Dict[str, DiagramElement] = field(default_factory=dict)


