fix: mypy errors

Author: angstwad

src/hydrator/cli.py

@@ -51,7 +51,8 @@ def hydration_worker(
         gatekeeper_validation: bool,
         gatekeeper_constraints: list[pathlib.Path],
         preserve_temp: bool,
-        split_output: bool
+        split_output: bool,
+        hydration_type: HydrateType
 ) -> HydrationResult:
     """
     Top-level function to be executed by each worker process.
@@ -100,6 +101,7 @@ def hydration_worker(
         validators=validators,
         preserve_temp=preserve_temp,
         split_output=split_output,
+        hydration_type=hydration_type
     )
 
     # Each worker runs its own asyncio event loop for the hydration task.
@@ -523,6 +525,7 @@ def _hydrate(self, config_data: SotConfig) -> None:
                     self._gatekeeper_constraints,
                     self._preserve_temp,
                     self._split_output,
+                    self._hyd_type,
                 )
                 futures[future] = cfg.name
 

src/hydrator/hydration.py

@@ -37,7 +37,7 @@ class BaseHydrator(LoggingMixin):
     """ Implements a specific hydration flow """
     _jinja_env: jinja2.Environment
     _hydration_dest: pathlib.Path
-    _oci_client: OCIClient
+    _oci_client: OCIClient | None
     _visited_files: Set[pathlib.Path]
 
     __slots__ = [
@@ -57,7 +57,7 @@ def __init__(self, *,
                  modules_path: pathlib.Path,
                  hydrated_path: pathlib.Path,
                  output_subdir: str,
-                 oci_client: OCIClient,
+                 oci_client: OCIClient | None,
                  oci_tags: Set[str],
                  hydration_type: HydrateType,
                  validators: list[BaseValidator] | None = None,
@@ -487,10 +487,11 @@ def _publish(self) -> None:
             self._logger,
         )
 
-        self._oci_client.push(
-            packaged_manifest_path,
-            self.name,
-            self.oci_tags)
+        if self._oci_client:
+            self._oci_client.push(
+                packaged_manifest_path,
+                self.name,
+                self.oci_tags)
 
     async def run(self) -> None:
         """Orchestrates the hydration process flow.