main.py

# Versor: Universal Geometric Algebra Neural Network
# Copyright (C) 2026 Eunkyum Kim <nemonanconcode@gmail.com>
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
#

"""Versor CLI entry point.

Dispatches geometric learning tasks via Hydra configuration.
"""

import hydra
from omegaconf import DictConfig

EXAMPLE_TASKS = {'manifold', 'hyperbolic', 'sanity'}

_TASK_MODULES = {
    'md17':      ('tasks.md17',                'MD17Task'),
    'sr':        ('tasks.symbolic_regression', 'SRTask'),
    'lqa':       ('tasks.lqa',                 'LQATask'),
    'deap_eeg':  ('tasks.deap_eeg',            'DEAPEEGTask'),
}

@hydra.main(version_base=None, config_path="conf", config_name="config")
def main(cfg: DictConfig):
    """Entry point for task execution. Delegates to specific task handlers.

    Args:
        cfg (DictConfig): Hydra configuration dictionary.
    """
    task_name = cfg.name

    if task_name in EXAMPLE_TASKS:
        raise ValueError(
            f"'{task_name}' is a synthetic example task. "
            f"Run it via: uv run python -m examples.main task={task_name}"
        )

    if task_name not in _TASK_MODULES:
        raise ValueError(f"Unknown task: {task_name}. Available: {list(_TASK_MODULES.keys())}")

    module_path, class_name = _TASK_MODULES[task_name]
    import importlib
    TaskClass = getattr(importlib.import_module(module_path), class_name)
    task = TaskClass(cfg)
    task.run()

if __name__ == "__main__":
    main()