diff options
| author | Lukas Kreussel <65088241+LLukas22@users.noreply.github.com> | 2023-09-17 23:11:01 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-09-17 22:11:01 +0100 |
| commit | 03e194123d743ca73b10797a315b8d47734735e8 (patch) | |
| tree | 66553a66f204fbfa9b85df0923a370966573ce7c /candle-pyo3/py_src/candle/__init__.py | |
| parent | c2b866172abaf1d4b8d75273c4f4e28a16d872f0 (diff) | |
| download | candle-03e194123d743ca73b10797a315b8d47734735e8.tar.gz candle-03e194123d743ca73b10797a315b8d47734735e8.tar.bz2 candle-03e194123d743ca73b10797a315b8d47734735e8.zip | |
Add return types to `*.pyi` stubs (#880)
* Start generating return types
* Finish tensor type hinting
* Add `save_gguf` to `utils`
* Typehint `quant-llama.py`
Diffstat (limited to 'candle-pyo3/py_src/candle/__init__.py')
| -rw-r--r-- | candle-pyo3/py_src/candle/__init__.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/candle-pyo3/py_src/candle/__init__.py b/candle-pyo3/py_src/candle/__init__.py index 49c96122..951609cc 100644 --- a/candle-pyo3/py_src/candle/__init__.py +++ b/candle-pyo3/py_src/candle/__init__.py @@ -1 +1,5 @@ -from .candle import *
\ No newline at end of file +from .candle import * + +__doc__ = candle.__doc__ +if hasattr(candle, "__all__"): + __all__ = candle.__all__
\ No newline at end of file |