Fix flake8 warnings

Author: MDUYN

pyindicators/__init__.py

@@ -1,4 +1,4 @@
-from .indicators import sma, rsi, is_crossover, crossunder, ema, wilders_rsi, \
+from .indicators import sma, rsi, crossunder, ema, wilders_rsi, \
     crossover, is_crossover, wma, macd, willr
 
 __all__ = [

pyindicators/indicators/crossover.py

@@ -9,12 +9,13 @@ def crossover(
     data: Union[PdDataFrame, PlDataFrame],
     first_column: str,
     second_column: str,
-    result_column = "crossover",
+    result_column="crossover",
     data_points: int = None,
     strict: bool = True,
 ) -> Union[PdDataFrame, PlDataFrame]:
     """
-    Identifies crossover points where `first_column` crosses above or below `second_column`.
+    Identifies crossover points where `first_column` crosses above
+    or below `second_column`.
 
     Args:
         data: Pandas or Polars DataFrame
@@ -32,15 +33,18 @@ def crossover(
 
     # Restrict data to the last `data_points` rows if specified
     if data_points is not None:
-        data = data.tail(data_points) if isinstance(data, PdDataFrame) else data.slice(-data_points)
+        data = data.tail(data_points) if isinstance(data, PdDataFrame) \
+            else data.slice(-data_points)
 
     # Pandas Implementation
     if isinstance(data, PdDataFrame):
         col1, col2 = data[first_column], data[second_column]
         prev_col1, prev_col2 = col1.shift(1), col2.shift(1)
 
         if strict:
-            crossover_mask = ((prev_col1 < prev_col2) & (col1 > col2)) | ((prev_col1 > prev_col2) & (col1 < col2))
+            crossover_mask = (
+                (prev_col1 < prev_col2)
+                & (col1 > col2)) | ((prev_col1 > prev_col2) & (col1 < col2))
         else:
             crossover_mask = (col1 > col2) | (col1 < col2)
 
@@ -52,12 +56,14 @@ def crossover(
         prev_col1, prev_col2 = col1.shift(1), col2.shift(1)
 
         if strict:
-            crossover_mask = ((prev_col1 < prev_col2) & (col1 > col2)) | ((prev_col1 > prev_col2) & (col1 < col2))
+            crossover_mask = ((prev_col1 < prev_col2) & (col1 > col2)) | \
+                ((prev_col1 > prev_col2) & (col1 < col2))
         else:
             crossover_mask = (col1 > col2) | (col1 < col2)
 
         # Convert boolean mask to 1s and 0s
-        data = data.with_columns(pl.when(crossover_mask).then(1).otherwise(0).alias(result_column))
+        data = data.with_columns(pl.when(crossover_mask).then(1)
+                                 .otherwise(0).alias(result_column))
 
     return data
 

pyindicators/indicators/macd.py

@@ -1,9 +1,7 @@
 from typing import Union
 
-import numpy as np
 from pandas import DataFrame as PdDataFrame
 from polars import DataFrame as PlDataFrame
-import pandas as pd
 import polars as pl
 
 from pyindicators.exceptions import PyIndicatorException
@@ -21,20 +19,27 @@ def macd(
     histogram_column: str = "macd_histogram"
 ) -> Union[PdDataFrame, PlDataFrame]:
     """
-    Calculate the MACD (Moving Average Convergence Divergence) for a given DataFrame.
+    Calculate the MACD (Moving Average Convergence Divergence) for
+    a given DataFrame.
 
     Args:
-        data (Union[pd.DataFrame, pl.DataFrame]): Input data containing the price series.
+        data (Union[pd.DataFrame, pl.DataFrame]): Input data containing
+            the price series.
         source_column (str): Column name for the price series.
-        short_period (int, optional): Period for the short-term EMA (default: 12).
-        long_period (int, optional): Period for the long-term EMA (default: 26).
-        signal_period (int, optional): Period for the Signal Line EMA (default: 9).
+        short_period (int, optional): Period for the short-term EMA
+            (default: 12).
+        long_period (int, optional): Period for the long-term EMA
+            (default: 26).
+        signal_period (int, optional): Period for the Signal Line
+            EMA (default: 9).
         macd_column (str, optional): Column name to store the MACD line.
         signal_column (str, optional): Column name to store the Signal line.
-        histogram_column (str, optional): Column name to store the MACD histogram.
+        histogram_column (str, optional): Column name to store the
+            MACD histogram.
 
     Returns:
-        Union[pd.DataFrame, pl.DataFrame]: DataFrame with MACD, Signal Line, and Histogram.
+        Union[pd.DataFrame, pl.DataFrame]: DataFrame with MACD, Signal
+            Line, and Histogram.
     """
     if source_column not in data.columns:
         raise PyIndicatorException(

pyindicators/indicators/williams_percent_range.py

@@ -49,8 +49,10 @@ def willr(
         return data.drop(columns=["high_n", "low_n"])
 
     elif isinstance(data, pl.DataFrame):
-        high_n = data.select(pl.col(high_column).rolling_max(period).alias("high_n"))
-        low_n = data.select(pl.col(low_column).rolling_min(period).alias("low_n"))
+        high_n = data.select(pl.col(high_column).rolling_max(period)
+                             .alias("high_n"))
+        low_n = data.select(pl.col(low_column).rolling_min(period)
+                            .alias("low_n"))
 
         data = data.with_columns([
             high_n["high_n"],