Remove warnings on CLI start (#502)

* add and test

* fix

* move warning removal to init logger

* revert logger changes

* remove revert again

Author: CTY-git

patchwork/common/utils/progress_bar.py

@@ -2,7 +2,6 @@
 
 import contextlib
 import functools
-import warnings
 from collections import Counter
 
 from rich.progress import Progress, SpinnerColumn, TimeElapsedColumn
@@ -18,7 +17,6 @@ class PatchflowProgressBar:
     def __init__(self, patchflow: Step):
         self.__step_counter = Counter()
         self.__current_progress = 0.00
-        self.__callbacks = []
         self.__patchflow_name = patchflow.__class__.__name__
 
         patchflow_run_func = patchflow.run
@@ -30,13 +28,9 @@ def inner_run():
                 self.__progress_bar_update(
                     description=f"[bold green]Finished {self.__patchflow_name}", completed=self.__MAX_PROGRESS
                 )
-                self.__do_callbacks()
 
         patchflow.run = inner_run
 
-    def register_callbacks(self, *callbacks):
-        self.__callbacks.extend(callbacks)
-
     def register_steps(self, *steps: Type[Step]):
         for step in steps:
             self.register_step(step)
@@ -69,7 +63,6 @@ def __progress_bar(self):
 
     @functools.cached_property
     def __progress_bar_update(self):
-        self.__suppress_warnings()
         progress = self.__progress_bar
         logger.register_progress_bar(progress)
         task_id = progress.add_task(
@@ -87,11 +80,3 @@ def __update(self, step: type):
         self.__step_counter[step] += 1
         yield
         return
-
-    def __suppress_warnings(self):
-        warnings.simplefilter("ignore")
-        self.__callbacks.append(warnings.resetwarnings)
-
-    def __do_callbacks(self):
-        for callbacks in self.__callbacks:
-            callbacks()

patchwork/logger.py

@@ -3,6 +3,7 @@
 import contextlib
 import logging
 import os
+import warnings
 from functools import partial
 
 import click
@@ -138,6 +139,7 @@ def inner(record: logging.LogRecord) -> bool:
 def init_cli_logger(log_level: str) -> logging.Logger:
     global logger, __noop
 
+    warnings.simplefilter("ignore")
     logger.removeHandler(__noop)
 
     if not os.path.exists(HOME_FOLDER):  # Check if HOME_FOLDER exists at this point

patchwork/steps/CallLLM/CallLLM.py

@@ -226,13 +226,11 @@ def run(self) -> dict:
             self.set_status(StepStatus.SKIPPED, "No prompts to process")
             return dict(openai_responses=[])
 
-        if prompt_length > self.call_limit:
+        if -1 < self.call_limit < prompt_length:
             logger.debug(
                 f"Number of prompts ({prompt_length}) exceeds the call limit ({self.call_limit}). "
                 f"Only the first {self.call_limit} prompts will be processed."
             )
-
-        if self.call_limit > 0:
             prompts = list(islice(self.prompts, self.call_limit))
         else:
             prompts = self.prompts

patchwork/steps/PreparePR/PreparePR.py

@@ -47,7 +47,7 @@ def run(self) -> dict:
                 placeholder_inner_text = path
                 # TODO: consider dealing with line numbers exceeding diff chunk
                 # if start_line is not None and end_line is not None:
-                    # placeholder_inner_text = f"{path}:{start_line+1}:{end_line}"
+                # placeholder_inner_text = f"{path}:{start_line+1}:{end_line}"
                 # chunk_link = "{{" + placeholder_inner_text + "}}"
 
                 if title != "" and patch_msg == "":