🎨 Format Python code with psf/black

Author: ndem0

ezyrb/database.py

@@ -42,7 +42,6 @@ def __init__(self, parameters=None, snapshots=None, space=None):
         )
         self._pairs = []
 
-
         if parameters is None and snapshots is None:
             logger.debug("Empty database created")
             return
@@ -146,7 +145,9 @@ def add(self, parameter, snapshot):
         """
         if not isinstance(parameter, Parameter):
             logger.error("Invalid parameter type: %s", type(parameter))
-            raise TypeError(f"Expected a Parameter object, got {type(parameter)}")
+            raise TypeError(
+                f"Expected a Parameter object, got {type(parameter)}"
+            )
 
         if not isinstance(snapshot, Snapshot):
             logger.error("Invalid snapshot type: %s", type(snapshot))
@@ -158,7 +159,7 @@ def add(self, parameter, snapshot):
         )
 
         return self
-    
+
     def split(self, chunks, seed=None):
         """
 
@@ -167,9 +168,7 @@ def split(self, chunks, seed=None):
         >>> train, test = db.split([80, 20])   # n snapshots
 
         """
-        logger.debug(
-            "Splitting database with chunks=%s, seed=%s", chunks, seed
-        )
+        logger.debug("Splitting database with chunks=%s, seed=%s", chunks, seed)
 
         if seed is not None:
             np.random.seed(seed)
@@ -206,7 +205,9 @@ def split(self, chunks, seed=None):
 
         else:
             logger.error("Invalid chunk type")
-            raise TypeError(f"Invalid chunk type. Expected a list of integers or floats, but got {type(chunks)}.")
+            raise TypeError(
+                f"Invalid chunk type. Expected a list of integers or floats, but got {type(chunks)}."
+            )
 
         new_database = [Database() for _ in range(len(chunks))]
         for i, chunk in enumerate(chunks):
@@ -232,4 +233,4 @@ def get_snapshot_space(self, index):
         """
         if index < 0 or index >= len(self._pairs):
             raise IndexError("Snapshot index out of range.")
-        return self._pairs[index][1].space
+        return self._pairs[index][1].space

ezyrb/plugin/scaler.py

@@ -61,7 +61,6 @@ def target(self):
         rtype: str
         """
         return self._target
-    
 
     @target.setter
     def target(self, new_target):
@@ -84,13 +83,14 @@ def mode(self):
     @mode.setter
     def mode(self, new_mode):
         if new_mode not in ["full", "reduced"]:
-            error_msg = f"Invalid mode: '{new_mode}' must be 'full' or 'reduced'."
+            error_msg = (
+                f"Invalid mode: '{new_mode}' must be 'full' or 'reduced'."
+            )
             logger.error(error_msg)
             raise ValueError(error_msg)
 
         self._mode = new_mode
 
-
     def _select_matrix(self, db):
         """
         Helper function to select the proper matrix to rescale.