Remove traceback from logs and refine retryable_error worker

FRINXio · Oct 29, 2024 · 99adda4 · 99adda4
1 parent d51bac9
commit 99adda4
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 7 deletions.
diff --git a/examples/simple_worker/workers/test_worker.py b/examples/simple_worker/workers/test_worker.py
@@ -231,13 +231,13 @@ def fail_three_times() -> None:
                 Simulates three consecutive failures by using an environment variable to track attempts.
                 Raises an exception on the first three executions and succeeds on the fourth.
                 """
-                env_var_name: str = f"{self.__class__.__name__}_attempt_count"
-                attempt_count: int = int(os.getenv(env_var_name, 0))
+                env_var_name: str = f'{self.__class__.__name__}_attempt_count'
+                attempt_count: int = int(os.getenv(env_var_name, 'O'))
 
                 try:
-                    if attempt_count < 3:
-                        os.environ[env_var_name]: str = str(attempt_count + 1)
-                        raise RuntimeError("Simulated failure")
+                    if attempt_count < 3:  # noqa: PLR2004
+                        os.environ[env_var_name] = str(attempt_count + 1)
+                        raise RuntimeError('Simulated failure')
                     else:
                         os.environ.pop(env_var_name, None)
                 except RuntimeError as e:

diff --git a/frinx/common/worker/worker.py b/frinx/common/worker/worker.py
@@ -243,8 +243,6 @@ def exception_response_handler(self, task: RawTaskIO, error: Exception, **kwargs
 
             task_result.output = TaskOutput(**error_dict_with_output_path)
 
-            logger.error('%s error occurred: %s \n%s', error_name, error, str(traceback.format_exc()))
-
         return task_result
 
     def execute_wrapper(self, task: RawTaskIO) -> Any: