Complete type hints coverage for Python code (#898)

pola-rs · Jun 29, 2021 · 6b80e50 · 6b80e50
1 parent e4f1806
commit 6b80e50
Show file tree

Hide file tree

Showing 6 changed files with 327 additions and 313 deletions.
diff --git a/py-polars/mypy.ini b/py-polars/mypy.ini
@@ -1,4 +1,4 @@
 [mypy]
 ignore_missing_imports = True
-disallow_untyped_defs = False
+disallow_untyped_defs = True
 files = polars
diff --git a/py-polars/polars/ffi.py b/py-polars/polars/ffi.py
@@ -29,7 +29,7 @@ def _ptr_to_numpy(ptr: int, len: int, ptr_type: Any) -> np.ndarray:
     return ctypeslib.as_array(ptr_ctype, (len,))
 
 
-def _as_float_ndarray(ptr, size):
+def _as_float_ndarray(ptr: int, size: int) -> np.ndarray:
     """
     https://github.com/maciejkula/python-rustlearn