Merge pull request #173 from ecmwf-projects/COPDS-1713-max-portal-costs

Introduce kwargs in estimate_costs

cads_adaptors/adaptors/__init__.py

@@ -145,14 +145,16 @@ def apply_constraints(self, request: Request) -> dict[str, Any]:
         pass
 
     @abc.abstractmethod
-    def estimate_costs(self, request: Request) -> dict[str, int]:
+    def estimate_costs(self, request: Request, **kwargs: Any) -> dict[str, int]:
         """
         Estimate the costs associated with the request.
 
         Parameters
         ----------
         request : Request
             Incoming request.
+        **kwargs : Any
+            Additional parameters, specific to the particular method's implementation.
 
         Returns
         -------
@@ -202,7 +204,7 @@ class DummyAdaptor(AbstractAdaptor):
     def apply_constraints(self, request: Request) -> dict[str, Any]:
         return {}
 
-    def estimate_costs(self, request: Request) -> dict[str, int]:
+    def estimate_costs(self, request: Request, **kwargs: Any) -> dict[str, int]:
         size = int(request.get("size", 0))
         time = int(request.get("time", 0.0))
         return {"size": size, "time": time}

cads_adaptors/adaptors/cds.py

@@ -90,9 +90,8 @@ def intersect_constraints(self, request: Request) -> list[Request]:
     def apply_mapping(self, request: Request) -> Request:
         return mapping.apply_mapping(request, self.mapping)
 
-    def estimate_costs(
-        self, request: Request, cost_threshold: str = "max_costs"
-    ) -> dict[str, int]:
+    def estimate_costs(self, request: Request, **kwargs: Any) -> dict[str, int]:
+        cost_threshold = kwargs.get("cost_threshold", "max_costs")
         costing_config: dict[str, Any] = self.config.get("costing", dict())
         costing_kwargs: dict[str, Any] = costing_config.get("costing_kwargs", dict())
         cost_threshold = (