diff options
Diffstat (limited to 'candle-pyo3/stub.py')
| -rw-r--r-- | candle-pyo3/stub.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/candle-pyo3/stub.py b/candle-pyo3/stub.py index 8e4318bc..336f674b 100644 --- a/candle-pyo3/stub.py +++ b/candle-pyo3/stub.py @@ -13,7 +13,7 @@ GENERATED_COMMENT = "# Generated content DO NOT EDIT\n" TYPING = """from typing import Any, Callable, Dict, List, Optional, Tuple, Union, Sequence from os import PathLike """ -CANDLE_SPECIFIC_TYPING = "from candle.typing import _ArrayLike, Device, Scalar, Index\n" +CANDLE_SPECIFIC_TYPING = "from candle.typing import _ArrayLike, Device, Scalar, Index, Shape\n" CANDLE_TENSOR_IMPORTS = "from candle import Tensor,DType,QTensor\n" RETURN_TYPE_MARKER = "&RETURNS&: " ADDITIONAL_TYPEHINTS = {} |