mypy error fix

Netcracker · Jan 24, 2025 · c54de99 · c54de99
1 parent 78f3a42
commit c54de99
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/kubemarine/kubernetes/__init__.py b/kubemarine/kubernetes/__init__.py
@@ -19,7 +19,7 @@
 import re
 import json
 from contextlib import contextmanager
-from typing import List, Dict, Iterator, Any, Optional
+from typing import List, Dict, Iterator, Any, Optional, Union
 
 import yaml
 from jinja2 import Template
@@ -1296,7 +1296,7 @@ def prepare_audit_policy(group: NodeGroup) -> None:
     # upload rules on cluster
     group.put(io.StringIO(policy_config_file), audit_file_name, sudo=True, backup=True)
 
-def migrate_kubeadm_config(group: NodeGroup, config_file: str) -> None:
+def migrate_kubeadm_config(group: Union[NodeGroup, DeferredGroup], config_file: str) -> None:
     """
     Check if migration is needed based on Kubernetes minor version and perform the migration.