repr

gautam-404 · Oct 31, 2023 · cbb0a01 · cbb0a01
1 parent e84858d
commit cbb0a01
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/PreWhitener/PreWhitener.py b/PreWhitener/PreWhitener.py
@@ -465,3 +465,8 @@ def remove_based_on_local_snr(self, df: pd.DataFrame, resolution: float = 3) ->
                     to_drop.append(i)
         return df.drop(index=to_drop)
 
+    def __repr__(self):
+        if self.name is not None:
+            return f"PreWhitener(name='{self.name}')"
+        else:
+            return f'PreWhitener(lc=({self.t}, {self.data}))'