mypy fix

Author: fatemetkl

examples/ensemble_attack_example/real_data_collection.py

@@ -8,9 +8,7 @@
 import pandas as pd
 from omegaconf import DictConfig
 
-from midst_toolkit.attacks.ensemble.data_utils import (
-    save_dataframe,
-)
+from midst_toolkit.attacks.ensemble.data_utils import save_dataframe
 
 
 def expand_ranges(ranges):

src/midst_toolkit/attacks/ensemble/process_split_data.py

@@ -5,9 +5,7 @@
 import pandas as pd
 from sklearn.model_selection import train_test_split
 
-from midst_toolkit.attacks.ensemble.data_utils import (
-    save_dataframe,
-)
+from midst_toolkit.attacks.ensemble.data_utils import save_dataframe
 from midst_toolkit.common.logger import log