better format

XanaduAI · Nov 14, 2024 · a63c66f · a63c66f
1 parent c57a72f
commit a63c66f
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 47 deletions.
diff --git a/mrmustard/physics/new_wires.py b/mrmustard/physics/new_wires.py
@@ -230,62 +230,22 @@ def __init__(
         self.classical_wires = set()
 
         for i, m in enumerate(sorted(modes_out_bra)):
-            self.quantum_wires.add(
-                QuantumWire(
-                    mode=m,
-                    is_out=True,
-                    is_ket=False,
-                    index=i,
-                )
-            )
+            self.quantum_wires.add(QuantumWire(mode=m, is_out=True, is_ket=False, index=i))
         n = len(modes_out_bra)
         for i, m in enumerate(sorted(modes_in_bra)):
-            self.quantum_wires.add(
-                QuantumWire(
-                    mode=m,
-                    is_out=False,
-                    is_ket=False,
-                    index=n + i,
-                )
-            )
+            self.quantum_wires.add(QuantumWire(mode=m, is_out=False, is_ket=False, index=n + i))
         n += len(modes_in_bra)
         for i, m in enumerate(sorted(modes_out_ket)):
-            self.quantum_wires.add(
-                QuantumWire(
-                    mode=m,
-                    is_out=True,
-                    is_ket=True,
-                    index=n + i,
-                )
-            )
+            self.quantum_wires.add(QuantumWire(mode=m, is_out=True, is_ket=True, index=n + i))
         n += len(modes_out_ket)
         for i, m in enumerate(sorted(modes_in_ket)):
-            self.quantum_wires.add(
-                QuantumWire(
-                    mode=m,
-                    is_out=False,
-                    is_ket=True,
-                    index=n + i,
-                )
-            )
+            self.quantum_wires.add(QuantumWire(mode=m, is_out=False, is_ket=True, index=n + i))
         n += len(modes_in_ket)
         for i, m in enumerate(sorted(classical_out)):
-            self.classical_wires.add(
-                ClassicalWire(
-                    mode=m,
-                    is_out=True,
-                    index=n + i,
-                )
-            )
+            self.classical_wires.add(ClassicalWire(mode=m, is_out=True, index=n + i))
         n += len(classical_out)
         for i, m in enumerate(sorted(classical_in)):
-            self.classical_wires.add(
-                ClassicalWire(
-                    mode=m,
-                    is_out=False,
-                    index=n + i,
-                )
-            )
+            self.classical_wires.add(ClassicalWire(mode=m, is_out=False, index=n + i))
 
     def copy(self) -> Wires:
         return deepcopy(self)

diff --git a/tests/test_physics/test_wires.py b/tests/test_physics/test_wires.py
@@ -21,7 +21,7 @@
 import pytest
 from ipywidgets import HTML
 
-from mrmustard.physics.new_wires import Wires, Repr
+from mrmustard.physics.new_wires import Repr, Wires
 
 
 class TestWires: