style: Format lattice.py with black

Author: Stellogic

tensorcircuit/templates/lattice.py

@@ -34,13 +34,13 @@ def __init__(self, dimensionality: int):
             dimensionality (int): The spatial dimension of the lattice (e.g., 1, 2, 3).
         """
         self._dimensionality = dimensionality
-        
+
         # --- Internal Data Structures (to be populated by subclasses) ---
         self._indices: List[SiteIndex] = []
         self._identifiers: List[SiteIdentifier] = []
         self._coordinates: List[Coordinates] = []
         self._ident_to_idx: Dict[SiteIdentifier, SiteIndex] = {}
-        
+
         # Neighbor information, structured as {k: {site_index: [neighbor_indices]}}
         # k=1 for nearest neighbors, k=2 for next-nearest, etc.
         self._neighbor_maps: Dict[int, NeighborMap] = {}
@@ -58,14 +58,14 @@ def dimensionality(self) -> int:
     def __len__(self) -> int:
         """Returns the total number of sites, enabling `len(lattice)`."""
         return self.num_sites
-    
+
     # Other common methods like get_site_info, get_neighbors can be added here.
 
     @abc.abstractmethod
     def _build_lattice(self, *args, **kwargs):
         """
         Abstract method for subclasses to generate the lattice data.
-        
+
         This method should populate the following internal lists and dicts:
         - self._indices
         - self._identifiers
@@ -90,7 +90,7 @@ def _build_neighbors(self, max_k: int = 1):
     def show(self, **kwargs):
         """
         Abstract method for visualizing the lattice structure.
-        
+
         Subclasses should implement the specific plotting logic.
         """
         pass
@@ -100,22 +100,30 @@ class GeneralLattice(AbstractLattice):
     """
     A general lattice built from an explicit list of sites and coordinates.
     """
-    def __init__(self,
-                 dimensionality: int,
-                 identifiers: List[SiteIdentifier],
-                 coordinates: List[Union[List[float], np.ndarray]]):
-        
+
+    def __init__(
+        self,
+        dimensionality: int,
+        identifiers: List[SiteIdentifier],
+        coordinates: List[Union[List[float], np.ndarray]],
+    ):
+
         super().__init__(dimensionality)
-        assert len(identifiers) == len(coordinates), "Identifiers and coordinates must have the same length."
+        assert len(identifiers) == len(
+            coordinates
+        ), "Identifiers and coordinates must have the same length."
 
         # The lattice is built directly upon initialization.
         self._build_lattice(identifiers=identifiers, coordinates=coordinates)
-        
+
         # Neighbor relationships can be calculated later if needed.
         print(f"GeneralLattice with {self.num_sites} sites created.")
 
-
-    def _build_lattice(self, identifiers: List[SiteIdentifier], coordinates: List[Union[List[float], np.ndarray]]):
+    def _build_lattice(
+        self,
+        identifiers: List[SiteIdentifier],
+        coordinates: List[Union[List[float], np.ndarray]],
+    ):
         """Implements the lattice building for GeneralLattice."""
         self._identifiers = list(identifiers)
         self._coordinates = [np.array(c) for c in coordinates]
@@ -131,7 +139,7 @@ def _build_neighbors(self, max_k: int = 1):
     def show(self, show_indices: bool = True, show_bonds: bool = False, **kwargs):
         """
         A simple visualization of the lattice sites.
-        
+
         Args:
             show_indices (bool): If True, display the identifier of each site.
             show_bonds (bool): If True, display lines connecting neighbors (not implemented yet).
@@ -148,19 +156,24 @@ def show(self, show_indices: bool = True, show_bonds: bool = False, **kwargs):
             return
 
         coords = np.array(self._coordinates)
-        
+
         plt.figure(figsize=(6, 6))
-        
+
         if coords.size > 0:
             plt.scatter(coords[:, 0], coords[:, 1], s=100, zorder=2)
 
         if show_indices:
             for i in range(self.num_sites):
-                plt.text(coords[i, 0] + 0.1, coords[i, 1] + 0.1, str(self._identifiers[i]), fontsize=12)
-        
-        plt.axis('equal')
+                plt.text(
+                    coords[i, 0] + 0.1,
+                    coords[i, 1] + 0.1,
+                    str(self._identifiers[i]),
+                    fontsize=12,
+                )
+
+        plt.axis("equal")
         plt.grid(True)
         plt.title(f"{self.__class__.__name__} ({self.num_sites} sites)")
         plt.xlabel("x coordinate")
         plt.ylabel("y coordinate")
-        plt.show()
+        plt.show()