fix: don't recover safety as inference

Haidra-Org · May 19, 2024 · d6e4743 · d6e4743
1 parent 6969607
commit d6e4743
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/horde_worker_regen/process_management/process_manager.py b/horde_worker_regen/process_management/process_manager.py
@@ -3742,7 +3742,10 @@ def _check_and_replace_process(
 
  if time_elapsed > timeout and process_info.last_process_state == state:
  logger.error(f"{process_info} {error_message}, replacing it")
- self._replace_inference_process(process_info)
+ if process_info.process_type == HordeProcessType.SAFETY:
+ self._replace_all_safety_process()
+ if process_info.process_type == HordeProcessType.INFERENCE:
+ self._replace_inference_process(process_info)
  return True
  return False