options.py

from .normalization import NORM2CLS
from .embeddings import EBB2CLS
from .activation import ACT2CLS
from .attention import ATTN2CLS
from .mlp import MLP2CLS
from .layers import LAYER2CLS


def print_layer_options():
    print("Activation options:")
    for option in ACT2CLS:
        print(f"  {option}")

    print("\nAttention Layer options:")
    for option in ATTN2CLS:
        print(f"  {option}")

    print("\nEmbedding Layer options:")
    for option in EBB2CLS:
        print(f"  {option}")

    print("\nBert Layer options:")
    for option in LAYER2CLS:
        print(f"  {option}")

    print("\nMLP Layer options:")
    for option in MLP2CLS:
        print(f"  {option}")

    print("\nNormalization options:")
    for option in NORM2CLS:
        print(f"  {option}")