Merge pull request #119 from jorenham/complete/scipy.odr

complete `scipy.odr`

README.md

@@ -88,7 +88,7 @@ pip install scipy-stubs
 | `linalg`      | :heavy_check_mark:    | :heavy_check_mark: | :heavy_check_mark: | :heavy_check_mark: | :full_moon:            |
 | ~`misc`~      | :heavy_check_mark:    | :heavy_check_mark: | :heavy_check_mark: | :heavy_check_mark: | :full_moon:            |
 | `ndimage`     | :heavy_check_mark:    | :heavy_check_mark: | :heavy_check_mark: | :heavy_check_mark: | :full_moon:            |
-| `odr`         | :heavy_check_mark:    | :heavy_check_mark: | :heavy_check_mark: | :heavy_check_mark: | :waxing_crescent_moon: |
+| `odr`         | :heavy_check_mark:    | :heavy_check_mark: | :heavy_check_mark: | :heavy_check_mark: | :full_moon:            |
 | `optimize`    | :heavy_check_mark:    | :heavy_check_mark: | :heavy_check_mark: | :heavy_check_mark: | :first_quarter_moon:   |
 | `signal`      | :heavy_check_mark:    | :heavy_check_mark: | :heavy_check_mark: | :heavy_check_mark: | :waxing_crescent_moon: |
 | `sparse`      | :heavy_check_mark:    | :heavy_check_mark: | :heavy_check_mark: | :heavy_check_mark: | :waxing_crescent_moon: |

scipy-stubs/odr/__init__.pyi

@@ -1,4 +1,4 @@
-from . import models as models, odrpack
+from . import models, odrpack
 from ._models import *
 from ._odrpack import *
 

scipy-stubs/odr/_add_newdocs.pyi

@@ -0,0 +1 @@
+# nothin to see here

scipy-stubs/odr/_models.pyi

@@ -1,25 +1,30 @@
-from typing import Final
+from typing import Final, type_check_only
 
 from numpy._typing import _ArrayLikeInt_co
 from ._odrpack import Model
 
 __all__ = ["Model", "exponential", "multilinear", "polynomial", "quadratic", "unilinear"]
 
-multilinear: Final[_MultilinearModel] = ...
-exponential: Final[_ExponentialModel] = ...
-unilinear: Final[_UnilinearModel] = ...
-quadratic: Final[_QuadraticModel] = ...
+@type_check_only
+class _NamedModel(Model):
+    name: Final[str]
+    equ: Final[str]
+    TeXequ: Final[str]
 
-class _MultilinearModel(Model):
+@type_check_only
+class _SimpleModel(_NamedModel):
     def __init__(self, /) -> None: ...
 
-class _ExponentialModel(Model):
-    def __init__(self, /) -> None: ...
+###
 
-class _UnilinearModel(Model):
-    def __init__(self, /) -> None: ...
+class _MultilinearModel(_SimpleModel): ...
+class _ExponentialModel(_SimpleModel): ...
+class _UnilinearModel(_SimpleModel): ...
+class _QuadraticModel(_SimpleModel): ...
 
-class _QuadraticModel(Model):
-    def __init__(self, /) -> None: ...
+def polynomial(order: _ArrayLikeInt_co) -> _NamedModel: ...
 
-def polynomial(order: _ArrayLikeInt_co) -> Model: ...
+multilinear: Final[_MultilinearModel] = ...
+exponential: Final[_ExponentialModel] = ...
+unilinear: Final[_UnilinearModel] = ...
+quadratic: Final[_QuadraticModel] = ...