From 4e5517e07700b856a1a3251bb5cb4553f3269ce7 Mon Sep 17 00:00:00 2001
From: Arsalan Motamedi <motamedi17@gmail.com>
Date: Fri, 31 Jan 2025 16:04:09 -0500
Subject: [PATCH] formatting changes

---
 tests/test_training/test_callbacks.py   |  2 +-
 tests/test_training/test_opt_lab_dev.py | 20 ++++++++++----------
 tests/test_training/test_trainer.py     |  2 +-
 3 files changed, 12 insertions(+), 12 deletions(-)

diff --git a/tests/test_training/test_callbacks.py b/tests/test_training/test_callbacks.py
index 02dbf6fb7..c1bfd9b27 100644
--- a/tests/test_training/test_callbacks.py
+++ b/tests/test_training/test_callbacks.py
@@ -18,7 +18,7 @@
 import tensorflow as tf
 
 from mrmustard import math, settings
-from mrmustard.lab_dev import Circuit, BSgate, S2gate, Vacuum
+from mrmustard.lab_dev import BSgate, Circuit, S2gate, Vacuum
 from mrmustard.training import Optimizer, TensorboardCallback
 
 from ..conftest import skip_np
diff --git a/tests/test_training/test_opt_lab_dev.py b/tests/test_training/test_opt_lab_dev.py
index 766dc5332..f83f18396 100644
--- a/tests/test_training/test_opt_lab_dev.py
+++ b/tests/test_training/test_opt_lab_dev.py
@@ -22,24 +22,24 @@
 
 from mrmustard import math, settings
 from mrmustard.lab_dev import (
-    Circuit,
-    Sgate,
-    S2gate,
-    Vacuum,
     BSgate,
-    Ggate,
-    Interferometer,
-    Rgate,
+    Circuit,
     Dgate,
-    RealInterferometer,
     DisplacedSqueezed,
-    SqueezedVacuum,
+    Ggate,
     GKet,
+    Interferometer,
     Number,
+    RealInterferometer,
+    Rgate,
+    S2gate,
+    Sgate,
+    SqueezedVacuum,
     TwoModeSqueezedVacuum,
+    Vacuum,
 )
 from mrmustard.math.parameters import Variable, update_euclidean
-from mrmustard.physics.gaussian import von_neumann_entropy, number_means
+from mrmustard.physics.gaussian import number_means, von_neumann_entropy
 from mrmustard.training import Optimizer
 from mrmustard.training.callbacks import Callback
 
diff --git a/tests/test_training/test_trainer.py b/tests/test_training/test_trainer.py
index fe10b2021..76902327f 100644
--- a/tests/test_training/test_trainer.py
+++ b/tests/test_training/test_trainer.py
@@ -32,7 +32,7 @@
 except ImportError:
     ray_available = False
 
-from mrmustard.lab_dev import Dgate, GKet, Ggate, Vacuum
+from mrmustard.lab_dev import Dgate, Ggate, GKet, Vacuum
 from mrmustard.training import Optimizer
 from mrmustard.training.trainer import map_trainer, train_device, update_pop