pytorch-lightning

1
# Copyright The Lightning AI team.
2
#
3
# Licensed under the Apache License, Version 2.0 (the "License");
4
# you may not use this file except in compliance with the License.
5
# You may obtain a copy of the License at
6
#
7
#     http://www.apache.org/licenses/LICENSE-2.0
8
#
9
# Unless required by applicable law or agreed to in writing, software
10
# distributed under the License is distributed on an "AS IS" BASIS,
11
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
# See the License for the specific language governing permissions and
13
# limitations under the License.
14

15
from time import time
16
from typing import List, Optional
17

18
import lightning.app
19
from lightning.app.core.queues import QueuingSystem
20
from lightning.app.runners.backends.backend import Backend
21

22

23
class DockerBackend(Backend):
24
    def resolve_url(self, app, base_url: Optional[str] = None) -> None:
25
        pass
26

27
    def stop_work(self, app: "lightning.app.LightningApp", work: "lightning.app.LightningWork") -> None:
28
        pass
29

30
    def __init__(self, entrypoint_file: str):
31
        super().__init__(entrypoint_file=entrypoint_file, queues=QueuingSystem.REDIS, queue_id=str(int(time())))
32

33
    def create_work(self, app, work):
34
        pass
35

36
    def update_work_statuses(self, works) -> None:
37
        pass
38

39
    def stop_all_works(self, works: List["lightning.app.LightningWork"]) -> None:
40
        pass
41