diff --git a/codeflash/api/aiservice.py b/codeflash/api/aiservice.py index 157bf24e6..b8bc9454b 100644 --- a/codeflash/api/aiservice.py +++ b/codeflash/api/aiservice.py @@ -832,6 +832,7 @@ def get_optimization_review( replay_tests: str, calling_fn_details: str, language: str = "python", + **_kwargs: Any, ) -> OptimizationReviewResult: """Compute the optimization review of current Pull Request. diff --git a/codeflash/api/cfapi.py b/codeflash/api/cfapi.py index 52e03c638..f7957fa0d 100644 --- a/codeflash/api/cfapi.py +++ b/codeflash/api/cfapi.py @@ -285,6 +285,7 @@ def create_staging( optimization_review: str = "", original_line_profiler: str | None = None, optimized_line_profiler: str | None = None, + **_kwargs: Any, # ignores the language argument TODO Hesham: staging for all langs ) -> Response: """Create a staging pull request, targeting the specified branch. (usually 'staging').