private void resetToNormalReplication() {
    try {
      logger.info("Restoring normal replication for disabled partitions");

      // Helps the mock replication engine
      Set<PartitionId> copyOfDisabledReplicationPartitions = new HashSet<>(disabledReplicationPartitions);
      // Enable replication for previously disabled partitions
      replicationEngine.controlReplicationForPartitions(copyOfDisabledReplicationPartitions, Collections.emptyList(),
          true);
      disabledReplicationPartitions.clear();
      replicationPrioritizationMetrics.updateDisabledReplicationPartitions(disabledReplicationPartitions);
      logger.debug("Disabling high priority partition flag isHighPriorityReplicationRunning={}",
          isHighPriorityReplicationRunning.get());
    } catch (Exception e) {
      logger.error("Error resetting to normal replication", e);
    }
  }
Resets replication to normal operation, enabling all partitions.
ambry-prioritization/src/main/java/com/github/ambry/replica/prioritization/ReplicationPrioritizationManager.java