image-match/python/py/Lib/site-packages/wandb/sdk/backend/backend.py

"""Backend - Send to internal process.

Manage backend.

"""

from __future__ import annotations

import logging
from typing import TYPE_CHECKING

from wandb.sdk.interface.interface import InterfaceBase

if TYPE_CHECKING:
    from wandb.sdk.lib.service import service_connection
    from wandb.sdk.wandb_settings import Settings

logger = logging.getLogger("wandb")


class Backend:
    interface: InterfaceBase | None

    _settings: Settings

    _done: bool

    _service: service_connection.ServiceConnection | None

    def __init__(
        self,
        settings: Settings,
        service: service_connection.ServiceConnection | None = None,
    ) -> None:
        self._done = False

        self.interface = None

        self._settings = settings
        self._service = service

    def ensure_launched(self) -> None:
        """Launch backend worker if not running."""
        assert self._settings.run_id
        assert self._service
        self.interface = self._service.make_interface(
            stream_id=self._settings.run_id,
        )

    def server_status(self) -> None:
        """Report server status."""

    def cleanup(self) -> None:
        # TODO: make _done atomic
        if self._done:
            return
        self._done = True
        if self.interface:
            self.interface.join()