def get_total_batches()

in code/run_eval_prm_trl.py [0:0]

• 2 lines of code
• 1 McCabe index (conditional complexity)

    def get_total_batches(self):
        """Return the total number of batches."""
        return (self.total_steps + self.batch_size - 1) // self.batch_size