Fix flake8 warnings

Author: MDUYN

pyindicators/indicators/average_true_range.py

@@ -47,8 +47,10 @@ def atr(
         # Polars version
         df = data.with_columns([
             (pl.col("High") - pl.col("Low")).alias("H_L"),
-            (pl.col("High") - pl.col(source_column).shift(1)).abs().alias("H_Cp"),
-            (pl.col("Low") - pl.col(source_column).shift(1)).abs().alias("L_Cp"),
+            (pl.col("High") -
+                pl.col(source_column).shift(1)).abs().alias("H_Cp"),
+            (pl.col("Low") -
+                pl.col(source_column).shift(1)).abs().alias("L_Cp"),
         ])
 
         # True Range = max of H-L, H-Cprev, L-Cprev

pyindicators/indicators/bollinger_bands.py

@@ -1,7 +1,6 @@
 from typing import Union
 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
 
@@ -42,7 +41,9 @@ def bollinger_bands(
         ])
 
     else:
-        raise PyIndicatorException("Input data must be a pandas or polars DataFrame.")
+        raise PyIndicatorException(
+            "Input data must be a pandas or polars DataFrame."
+        )
 
 
 def bollinger_width(
@@ -71,12 +72,15 @@ def bollinger_width(
     if isinstance(data, PdDataFrame):
         data[result_column] = data['BB_upper_temp'] - data['BB_lower_temp']
         # Drop temporary columns
-        data = data.drop(columns=['BB_middle_temp', 'BB_upper_temp', 'BB_lower_temp'])
+        data = data.drop(
+            columns=['BB_middle_temp', 'BB_upper_temp', 'BB_lower_temp']
+        )
         return data
 
     elif isinstance(data, PlDataFrame):
         return data.with_columns(
-            (pl.col('BB_upper_temp') - pl.col('BB_lower_temp')).alias(result_column)
+            (pl.col('BB_upper_temp') -
+             pl.col('BB_lower_temp')).alias(result_column)
         ).drop(['BB_middle_temp', 'BB_upper_temp', 'BB_lower_temp'])
 
     else:

pyindicators/indicators/divergence.py

@@ -286,7 +286,8 @@ def bullish_divergence(
             or second_column_lows not in data.columns:
 
         # Check if the two columns are in the data
-        if first_column not in data.columns or second_column not in data.columns:
+        if first_column not in data.columns \
+                or second_column not in data.columns:
             raise PyIndicatorException(
                 f"{first_column} and {second_column} columns "
                 "are required in the data"
@@ -412,7 +413,8 @@ def bearish_divergence(
             or second_column_highs not in data.columns:
 
         # Check if the two columns are in the data
-        if first_column not in data.columns or second_column not in data.columns:
+        if first_column not in data.columns \
+                or second_column not in data.columns:
             raise PyIndicatorException(
                 f"{first_column} and {second_column} columns "
                 "are required in the data"
@@ -464,6 +466,7 @@ def bearish_divergence(
     df[result_column] = result
     return pl.DataFrame(df) if is_polars else df
 
+
 def bearish_divergence_multi_dataframe(
     first_df: Union[pd.DataFrame, pl.DataFrame],
     second_df: Union[pd.DataFrame, pl.DataFrame],