2
2
3
3
import asyncio
4
4
import os
5
- from typing import TYPE_CHECKING
5
+ from typing import TYPE_CHECKING , Any
6
6
from kubernetes .utils import parse_quantity
7
7
8
8
import yaml
@@ -44,7 +44,7 @@ def __init__(self):
44
44
self .compute_crawler_resources ()
45
45
self .compute_profile_resources ()
46
46
47
- def compute_crawler_resources (self ):
47
+ def compute_crawler_resources (self ) -> None :
48
48
"""compute memory / cpu resources for crawlers"""
49
49
p = self .shared_params
50
50
num_workers = max (int (p ["crawler_browser_instances" ]), 1 )
@@ -105,7 +105,7 @@ def compute_crawler_resources(self):
105
105
p ["qa_memory" ] = qa_memory
106
106
p ["qa_workers" ] = qa_num_workers
107
107
108
- def compute_profile_resources (self ):
108
+ def compute_profile_resources (self ) -> None :
109
109
"""compute memory /cpu resources for a single profile browser"""
110
110
p = self .shared_params
111
111
# if no profile specific options provided, default to crawler base for one browser
@@ -122,7 +122,7 @@ def compute_profile_resources(self):
122
122
print (f"cpu = { profile_cpu } " )
123
123
print (f"memory = { profile_memory } " )
124
124
125
- async def async_init (self ):
125
+ async def async_init (self ) -> None :
126
126
"""perform any async init here"""
127
127
self .has_pod_metrics = await self .is_pod_metrics_available ()
128
128
print ("Pod Metrics Available:" , self .has_pod_metrics )
@@ -172,16 +172,16 @@ def __init__(
172
172
# see: https://stackoverflow.com/a/74059981
173
173
self .bg_tasks = set ()
174
174
175
- def init_routes (self , app ):
175
+ def init_routes (self , app ) -> None :
176
176
"""init routes for this operator"""
177
177
178
- def run_task (self , func ):
178
+ def run_task (self , func ) -> None :
179
179
"""add bg tasks to set to avoid premature garbage collection"""
180
180
task = asyncio .create_task (func )
181
181
self .bg_tasks .add (task )
182
182
task .add_done_callback (self .bg_tasks .discard )
183
183
184
- def load_from_yaml (self , filename , params ):
184
+ def load_from_yaml (self , filename , params ) -> list [ Any ] :
185
185
"""load and parse k8s template from yaml file"""
186
186
return list (
187
187
yaml .safe_load_all (
0 commit comments