from __future__ import annotations

"""Toy Habitat Supervisor Adapter.

Skeleton adapter that would interface with NebulaForge on-device runtimes and
the EnergiBridge to coordinate habitat-level supervision tasks.
"""

from nebulaforge.runtime import DeviceRuntime


class HabitatSupervisorAdapter:
    def __init__(self, device: str = "arm-raspberrypi"):  # pragma: no cover
        self.runtime = DeviceRuntime(device=device)

    def initialize(self, config: dict | None = None) -> None:
        self.runtime.initialize(config)

    def infer(self, inputs):
        return self.runtime.infer(inputs)

    def plan(self, state):
        return self.runtime.plan(state)