File tree Expand file tree Collapse file tree 1 file changed +3
-6
lines changed
src/lightning_app/components/serve Expand file tree Collapse file tree 1 file changed +3
-6
lines changed Original file line number Diff line number Diff line change 7
7
8
8
import uvicorn
9
9
from fastapi import FastAPI
10
- from lightning_utilities .core .imports import compare_version
10
+ from lightning_utilities .core .imports import compare_version , module_available
11
11
from pydantic import BaseModel
12
12
13
13
from lightning_app .core .work import LightningWork
16
16
17
17
logger = Logger (__name__ )
18
18
19
- __doctest_skip__ = ["PythonServer" , "PythonServer.*" ]
20
-
21
-
22
19
# Skip doctests if requirements aren't available
23
- if not _is_torch_available ():
24
- __doctest_skip__ + = ["PythonServer" , "PythonServer.*" ]
20
+ if not module_available ( "lightning_api_access" ) or not _is_torch_available ():
21
+ __doctest_skip__ = ["PythonServer" , "PythonServer.*" ]
25
22
26
23
27
24
def _get_device ():
You can’t perform that action at this time.
0 commit comments