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 77
88import uvicorn
99from fastapi import FastAPI
10- from lightning_utilities .core .imports import compare_version
10+ from lightning_utilities .core .imports import compare_version , module_available
1111from pydantic import BaseModel
1212
1313from lightning_app .core .work import LightningWork
1616
1717logger = Logger (__name__ )
1818
19- __doctest_skip__ = ["PythonServer" , "PythonServer.*" ]
20-
21-
2219# 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.*" ]
2522
2623
2724def _get_device ():
You can’t perform that action at this time.
0 commit comments