make format

matsengrp · Jun 13, 2024 · edcfbea · edcfbea
1 parent 2c11a3f
commit edcfbea
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/netam/framework.py b/netam/framework.py
@@ -725,7 +725,9 @@ def joint_train(
         optimize_branch_lengths()
         self.mark_branch_lengths_optimized(0)
         for cycle in range(cycle_count):
-            print(f"### Beginning cycle {cycle + 1}/{cycle_count} using optimizer {self.optimizer_name}")
+            print(
+                f"### Beginning cycle {cycle + 1}/{cycle_count} using optimizer {self.optimizer_name}"
+            )
             self.mark_branch_lengths_optimized(cycle + 1)
             current_lr = self.optimizer.param_groups[0]["lr"]
             # set new_lr to be the geometric mean of current_lr and the