diff options
| author | Lukas Kreussel <65088241+LLukas22@users.noreply.github.com> | 2023-09-16 18:23:38 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-09-16 17:23:38 +0100 |
| commit | 8658df348527cabcd722bfe2e9e48aba3c7f8e96 (patch) | |
| tree | 5e7a04ad7650b7872074fee2197eb7db7e248bbb /candle-pyo3/py_src/candle/typing | |
| parent | 7cafca835a4bb9a21f3c8111e2f61b7a6b1270fd (diff) | |
| download | candle-8658df348527cabcd722bfe2e9e48aba3c7f8e96.tar.gz candle-8658df348527cabcd722bfe2e9e48aba3c7f8e96.tar.bz2 candle-8658df348527cabcd722bfe2e9e48aba3c7f8e96.zip | |
Generate `*.pyi` stubs for PyO3 wrapper (#870)
* Begin to generate typehints.
* generate correct stubs
* Correctly include stubs
* Add comments and typhints to static functions
* ensure candle-pyo3 directory
* Make `llama.rope.freq_base` optional
* `fmt`
Diffstat (limited to 'candle-pyo3/py_src/candle/typing')
| -rw-r--r-- | candle-pyo3/py_src/candle/typing/__init__.py | 16 |
1 files changed, 16 insertions, 0 deletions
diff --git a/candle-pyo3/py_src/candle/typing/__init__.py b/candle-pyo3/py_src/candle/typing/__init__.py new file mode 100644 index 00000000..ea85d2a3 --- /dev/null +++ b/candle-pyo3/py_src/candle/typing/__init__.py @@ -0,0 +1,16 @@ +from typing import TypeVar, Union, Sequence + +_T = TypeVar("_T") + +_ArrayLike = Union[ + _T, + Sequence[_T], + Sequence[Sequence[_T]], + Sequence[Sequence[Sequence[_T]]], + Sequence[Sequence[Sequence[Sequence[_T]]]], +] + +CPU:str = "cpu" +CUDA:str = "cuda" + +Device = TypeVar("Device", CPU, CUDA)
\ No newline at end of file |