pytorch-lightning

1
from enum import Enum
2

3
from lightning.app.core.constants import APP_SERVER_IN_CLOUD
4
from lightning.app.runners.backends.backend import Backend
5
from lightning.app.runners.backends.cloud import CloudBackend
6
from lightning.app.runners.backends.docker import DockerBackend
7
from lightning.app.runners.backends.mp_process import CloudMultiProcessingBackend, MultiProcessingBackend
8

9

10
class BackendType(Enum):
11
    MULTIPROCESSING = "multiprocessing"
12
    DOCKER = "docker"
13
    CLOUD = "cloud"
14

15
    def get_backend(self, entrypoint_file: str) -> "Backend":
16
        if self == BackendType.MULTIPROCESSING:
17
            if APP_SERVER_IN_CLOUD:
18
                return CloudMultiProcessingBackend(entrypoint_file)
19
            return MultiProcessingBackend(entrypoint_file)
20
        if self == BackendType.DOCKER:
21
            return DockerBackend(entrypoint_file)
22
        if self == BackendType.CLOUD:
23
            return CloudBackend(entrypoint_file)
24
        raise ValueError("Unknown client type")
25