check_untyped_defs pandas.io.sas.sasreader

Author: simonjayhawkins

pandas/io/sas/sasreader.py

@@ -1,16 +1,24 @@
 """
 Read SAS sas7bdat or xport files.
 """
+from typing import TYPE_CHECKING, AnyStr, Optional, Union
+
+from pandas._typing import FilePathOrBuffer
+
 from pandas.io.common import _stringify_path
 
+if TYPE_CHECKING:
+    from pandas.io.sas.sas_xport import XportReader  # noqa: F401
+    from pandas.io.sas.sas7bdat import SAS7BDATReader  # noqa: F401
+
 
 def read_sas(
-    filepath_or_buffer,
-    format=None,
+    filepath_or_buffer: FilePathOrBuffer[AnyStr],
+    format: Optional[str] = None,
     index=None,
-    encoding=None,
-    chunksize=None,
-    iterator=False,
+    encoding: Optional[str] = None,
+    chunksize: Optional[int] = None,
+    iterator: bool = False,
 ):
     """
     Read SAS files stored as either XPORT or SAS7BDAT format files.
@@ -63,14 +71,15 @@ def read_sas(
         else:
             raise ValueError("unable to infer format of SAS file")
 
+    reader: Union["XportReader", "SAS7BDATReader"]
     if format.lower() == "xport":
-        from pandas.io.sas.sas_xport import XportReader
+        from pandas.io.sas.sas_xport import XportReader  # noqa: F811
 
         reader = XportReader(
             filepath_or_buffer, index=index, encoding=encoding, chunksize=chunksize
         )
     elif format.lower() == "sas7bdat":
-        from pandas.io.sas.sas7bdat import SAS7BDATReader
+        from pandas.io.sas.sas7bdat import SAS7BDATReader  # noqa: F811
 
         reader = SAS7BDATReader(
             filepath_or_buffer, index=index, encoding=encoding, chunksize=chunksize

setup.cfg

@@ -309,9 +309,6 @@ check_untyped_defs=False
 [mypy-pandas.io.sas.sas7bdat]
 check_untyped_defs=False
 
-[mypy-pandas.io.sas.sasreader]
-check_untyped_defs=False
-
 [mypy-pandas.io.sql]
 check_untyped_defs=False