Typing: Narrow down types of arguments (NDFrame) pandas-dev#10 (pandas-dev#52754)

* improved specificationfor date_unit in to_json more precisely

* improved specification mode in to_hdf

* added the Literal to complib of to_hdf

* improved format for to_hdf

* address PR review

* Update generic.py

removed the new line in the import

* ran "pre-commit" to fix trailing whitespace

Author: mKlepsch

pandas/_typing.py

@@ -419,6 +419,7 @@ def closed(self) -> bool:
 AlignJoin = Literal["outer", "inner", "left", "right"]
 DtypeBackend = Literal["pyarrow", "numpy_nullable"]
 
+TimeUnit = Literal["s", "ms", "us", "ns"]
 OpenFileErrors = Literal[
     "strict",
     "ignore",

pandas/core/generic.py

@@ -82,6 +82,7 @@
     TimedeltaConvertibleTypes,
     TimeNonexistent,
     TimestampConvertibleTypes,
+    TimeUnit,
     ValueKeyFunc,
     WriteBuffer,
     WriteExcelBuffer,
@@ -2284,7 +2285,7 @@ def to_json(
         date_format: str | None = None,
         double_precision: int = 10,
         force_ascii: bool_t = True,
-        date_unit: str = "ms",
+        date_unit: TimeUnit = "ms",
         default_handler: Callable[[Any], JSONSerializable] | None = None,
         lines: bool_t = False,
         compression: CompressionOptions = "infer",
@@ -2564,11 +2565,11 @@ def to_hdf(
         self,
         path_or_buf: FilePath | HDFStore,
         key: str,
-        mode: str = "a",
+        mode: Literal["a", "w", "r+"] = "a",
         complevel: int | None = None,
-        complib: str | None = None,
+        complib: Literal["zlib", "lzo", "bzip2", "blosc"] | None = None,
         append: bool_t = False,
-        format: str | None = None,
+        format: Literal["fixed", "table"] | None = None,
         index: bool_t = True,
         min_itemsize: int | dict[str, int] | None = None,
         nan_rep=None,