/
__init__.py
63 lines (53 loc) · 1.65 KB
/
__init__.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
import typing as t
from typing import TYPE_CHECKING
if TYPE_CHECKING:
from typing import Literal
F = t.Callable[..., t.Any]
from pandas import Series as PdSeries
from pandas import DataFrame as PdDataFrame
from pandas._typing import Dtype as PdDType
from pandas._typing import DtypeArg as PdDTypeArg
from pyarrow.plasma import ObjectID
from pyarrow.plasma import PlasmaClient
DataFrameOrient = Literal["split", "records", "index", "columns", "values", "table"]
SeriesOrient = Literal["split", "records", "index", "table"]
# numpy is always required by bentoml
from numpy import generic as NpGeneric
from numpy.typing import NDArray as _NDArray
from numpy.typing import DTypeLike as NpDTypeLike
NpNDArray = _NDArray[t.Any]
from xgboost import DMatrix
from catboost import Pool as CbPool
from .starlette import ASGIApp
from .starlette import ASGISend
from .starlette import ASGIScope
from .starlette import ASGIMessage
from .starlette import ASGIReceive
from .starlette import AsgiMiddleware
WSGIApp = t.Callable[[F, t.Mapping[str, t.Any]], t.Iterable[bytes]]
__all__ = [
"PdSeries",
"PdDataFrame",
"PdDType",
"PdDTypeArg",
"DataFrameOrient",
"SeriesOrient",
"ObjectID",
"PlasmaClient",
# xgboost
"DMatrix",
"CbPool",
# numpy
"NpNDArray",
"NpGeneric",
"NpDTypeLike",
# starlette
"AsgiMiddleware",
"ASGIApp",
"ASGIScope",
"ASGISend",
"ASGIReceive",
"ASGIMessage",
# misc
"WSGIApp",
]