Lightning: make type hints public (#17100)

* Add missing MANIFESTs * move * one more * Ignore version.info properly * move * manifest --------- Co-authored-by: Carlos Mocholí <[email protected]> Co-authored-by: Jirka Borovec <[email protected]> Co-authored-by: Jirka <[email protected]> Co-authored-by: Adrian Wälchli <[email protected]>
Lightning-AI · Apr 27, 2023 · 3d7360a · 3d7360a
1 parent 8c18439
commit 3d7360a
Show file tree

Hide file tree

Showing 7 changed files with 11 additions and 5 deletions.
diff --git a/.gitignore b/.gitignore
@@ -53,17 +53,17 @@ wheels/
 .installed.cfg
 *.egg
 src/*/version.info
-src/lightning_app/
-src/lightning_fabric/
-src/pytorch_lightning/
+src/lightning_app/*
+src/lightning_fabric/*
+src/pytorch_lightning/*
 !src/*/__about__.py
 !src/*/__main__.py
 !src/*/__setup__.py
 !src/*/__version__.py
 !src/*/MANIFEST.in
 !src/*/py.typed
 !src/*/README.md
-!src/*/*.info
+!src/*/shell-folder_code-lives-lightning.info
 
 # PyInstaller
 #  Usually these files are written by a python script from a template

diff --git a/setup.py b/setup.py
@@ -87,7 +87,11 @@ def _set_manifest_path(manifest_dir: str, aggregate: bool = False, mapping: Mapp
     if aggregate:
         # aggregate all MANIFEST.in contents into a single temporary file
         manifest_path = _named_temporary_file(manifest_dir)
-        lines = ["include src/lightning/version.info\n", "include requirements/base.txt\n"]
+        lines = [
+            "include src/lightning/version.info\n",
+            "include src/lightning/py.typed\n",
+            "include requirements/base.txt\n",
+        ]
         # load manifest and aggregated all manifests
         for pkg in mapping.values():
             pkg_manifest = os.path.join(_PATH_SRC, pkg, "MANIFEST.in")

diff --git a/src/lightning/py.typed b/src/lightning/py.typed
diff --git a/src/lightning_app/MANIFEST.in b/src/lightning_app/MANIFEST.in
@@ -8,3 +8,4 @@ recursive-include src/lightning_app/cli/*-template *
 # TODO: remove this once lightning-ui package is ready as a dependency
 recursive-include src/lightning_app/ui *
 include src/lightning_app/components/serve/catimage.png
+include src/lightning_app/py.typed  # marker file for PEP 561
diff --git a/src/lightning_app/py.typed b/src/lightning_app/py.typed
diff --git a/src/lightning_fabric/MANIFEST.in b/src/lightning_fabric/MANIFEST.in
@@ -4,3 +4,4 @@ include src/lightning_fabric/CHANGELOG.md
 include src/lightning_fabric/README.md
 recursive-include requirements/fabric *.txt
 include .actions/assistant.py
+include src/lightning_fabric/py.typed  # marker file for PEP 561
diff --git a/src/lightning_fabric/py.typed b/src/lightning_fabric/py.typed