1- import logging
2- from typing import Optional , Union
1+ from logging import getLogger
2+ from typing import List , Optional , Union
33
44from firebolt .common .util import prune_dict
55from firebolt .model import FireboltBaseModel
1212from firebolt .service .base import BaseService
1313from firebolt .service .types import EngineOrder , EngineType , WarmupMethod
1414
15- logger = logging . getLogger (__name__ )
15+ logger = getLogger (__name__ )
1616
1717
1818class EngineService (BaseService ):
@@ -24,10 +24,10 @@ def get(self, id_: str) -> Engine:
2424 engine_entry : dict = response .json ()["engine" ]
2525 return Engine .parse_obj_with_service (obj = engine_entry , engine_service = self )
2626
27- def get_by_ids (self , ids : list [str ]) -> list [Engine ]:
27+ def get_by_ids (self , ids : List [str ]) -> List [Engine ]:
2828 """Get multiple Engines from Firebolt by their ids."""
2929 response = self .client .post (
30- url = f "/core/v1/engines:getByIds" ,
30+ url = "/core/v1/engines:getByIds" ,
3131 json = {
3232 "engine_ids" : [
3333 {"account_id" : self .account_id , "engine_id" : engine_id }
@@ -56,7 +56,7 @@ def get_many(
5656 current_status_not_eq : str ,
5757 region_eq : str ,
5858 order_by : Union [str , EngineOrder ],
59- ) -> list [Engine ]:
59+ ) -> List [Engine ]:
6060 """
6161 Get a list of engines on Firebolt.
6262
@@ -73,7 +73,7 @@ def get_many(
7373 if isinstance (order_by , str ):
7474 order_by = EngineOrder [order_by ]
7575 response = self .client .get (
76- url = f "/core/v1/account/engines" ,
76+ url = "/core/v1/account/engines" ,
7777 params = prune_dict (
7878 {
7979 "page.first" : 5000 , # FUTURE: pagination support w/ generator
0 commit comments