Fix full packages

youtype · Dec 26, 2024 · e769f36 · e769f36
1 parent 81ab939
commit e769f36
Show file tree

Hide file tree

Showing 4 changed files with 21 additions and 5 deletions.
diff --git a/.github/workflows/integration_full.yml b/.github/workflows/integration_full.yml
@@ -26,8 +26,8 @@ jobs:
           - "types-aiobotocore-custom"
         python_version:
           - "3.8"
-          # - "3.10"
-          # - "3.13"
+          - "3.10"
+          - "3.13"
     steps:
       - uses: actions/checkout@v4
       - name: Set up uv
@@ -39,4 +39,5 @@ jobs:
           uv run scripts/integration.py \
             --product ${{ matrix.product }} \
             --python ${{ matrix.python_version }} \
+            --wheel \
             -d
diff --git a/mypy_boto3_builder/generators/aiobotocore_generator.py b/mypy_boto3_builder/generators/aiobotocore_generator.py
@@ -152,11 +152,16 @@ def generate_full_stubs(self) -> Package | None:
                 "*.pyi",
             )
         package.set_description(package.get_short_description("All-in-one type annotations"))
-        self._generate_full_stubs_services(package)
         self.package_writer.write_package(
             package,
             template_path=TemplatePath.types_aiobotocore_full,
         )
+        self._generate_full_stubs_services(package)
+        self.setup_package_writer.write_package(
+            package,
+            template_path=TemplatePath.types_aiobotocore_full,
+            include_template_names=("setup.py.jinja2",),
+        )
         return package
 
     def generate_custom_stubs(self) -> Package:

diff --git a/mypy_boto3_builder/generators/boto3_generator.py b/mypy_boto3_builder/generators/boto3_generator.py
@@ -192,12 +192,17 @@ def generate_full_stubs(self) -> Package | None:
                 "*.pyi",
             )
         package.set_description(package.get_short_description("All-in-one type annotations"))
-        self._generate_full_stubs_services(package)
         self.package_writer.write_package(
             package=package,
             template_path=TemplatePath.types_boto3_full,
             static_files_path=None,
         )
+        self._generate_full_stubs_services(package)
+        self.setup_package_writer.write_package(
+            package,
+            template_path=TemplatePath.types_boto3_full,
+            include_template_names=("setup.py.jinja2",),
+        )
         return package
 
     def generate_custom_stubs(self) -> Package:

diff --git a/mypy_boto3_builder/generators/types_boto3_generator.py b/mypy_boto3_builder/generators/types_boto3_generator.py
@@ -155,12 +155,17 @@ def generate_full_stubs(self) -> Package | None:
                 "*.pyi",
             )
         package.set_description(package.get_short_description("All-in-one type annotations"))
-        self._generate_full_stubs_services(package)
         self.package_writer.write_package(
             package=package,
             template_path=TemplatePath.types_boto3_full,
             static_files_path=None,
         )
+        self._generate_full_stubs_services(package)
+        self.setup_package_writer.write_package(
+            package,
+            template_path=TemplatePath.types_boto3_full,
+            include_template_names=("setup.py.jinja2",),
+        )
         return package
 
     def generate_custom_stubs(self) -> Package: