MAINT, TYP: stats: add __init__.pyi and ignore errors in test_samplin…

…g.py Stubs were not getting exported as no __init__.pyi file was found by mypy. This file has been added and other errors (false positives) have been ignored.
tirthasheshpatel · Jun 18, 2021 · 0a9386d · 0a9386d
1 parent 6ae6b79
commit 0a9386d
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 1 deletion.
diff --git a/mypy.ini b/mypy.ini
@@ -393,6 +393,9 @@ ignore_errors = True
 [mypy-scipy.stats.tests.test_qmc]
 ignore_errors = True
 
+[mypy-scipy.stats.tests.test_sampling]
+ignore_errors = True
+
 #
 # Files referencing compiled code that needs stubs added.
 #

diff --git a/scipy/stats/unuran/__init__.py b/scipy/stats/unuran/__init__.py
@@ -1 +1,4 @@
-from .unuran_wrapper import *
+from .unuran_wrapper import (
+    TransformedDensityRejection,
+    DiscreteAliasUrn
+)
diff --git a/scipy/stats/unuran/__init__.pyi b/scipy/stats/unuran/__init__.pyi
@@ -0,0 +1,4 @@
+from .unuran_wrapper import (
+    TransformedDensityRejection as TransformedDensityRejection,
+    DiscreteAliasUrn as DiscreteAliasUrn
+)