Produce py.typed and reexports for mypy

Author: nx10

src/styx/backend/python/languageprovider.py

@@ -1015,7 +1015,7 @@ def _not_is_instance(pytype: str):
                             *_assert_dict(get_param_or_die),
                             *_assert_tagged(get_param_or_die),
                             *_check_error(
-                                f"{get_param_or_die}[\"@type\"] not in {self.expr_literal(valid_tags)}",
+                                f'{get_param_or_die}["@type"] not in {self.expr_literal(valid_tags)}',
                                 f"Parameter `{p.base.name}`s `@type` must be one of {self.expr_literal(valid_tags)}",
                             ),
                             expr_validate,
@@ -1154,11 +1154,19 @@ def _get_checked(store: dict[str, typing.Any], key: str, type_: type[_T]) -> _T
                 path=python_package_path_src / f"{project.name}_{package.name}" / package_symbol / "__init__.py",
                 content=collapse(self.generate_module(package_module)),
             )
+            yield TextFile(
+                path=python_package_path_src / f"{project.name}_{package.name}" / "py.typed",
+                content="",
+            )
 
         yield TextFile(
             path=pathlib.Path(project.name) / f"src/{project.name}/__init__.py",
             content=template_root_init_py(project, package_names),
         )
+        yield TextFile(
+            path=pathlib.Path(project.name) / f"src/{project.name}/py.typed",
+            content="",
+        )
 
         yield TextFile(
             path=pathlib.Path(project.name) / f"pyproject.toml",

src/styx/backend/python/templates.py

@@ -78,10 +78,22 @@ def template_root_init_py(project: ir.Project, package_names: list[str]) -> str:
     ])
 
     return f'''{reexports}
-from styxdefs import *  # Reexport styxdefs
-from styxdocker import DockerRunner
-from styxsingularity import SingularityRunner
-from styxgraph import GraphRunner
+from styxdefs import (
+    Execution as Execution,
+    InputPathType as InputPathType,
+    Metadata as Metadata,
+    OutputPathType as OutputPathType,
+    Runner as Runner,
+    LocalRunner as LocalRunner,
+    DryRunner as DryRunner,
+    get_global_runner as get_global_runner,
+    set_global_runner as set_global_runner,
+    StyxRuntimeError as StyxRuntimeError,
+    StyxValidationError as StyxValidationError,
+)
+from styxdocker import DockerRunner as DockerRunner
+from styxsingularity import SingularityRunner as SingularityRunner
+from styxgraph import GraphRunner as GraphRunner
 
 
 def use_local(*args, **kwargs):