mypy fixes

Author: fatemetkl

examples/ensemble_attack_example/real_data_collection.py

@@ -8,7 +8,7 @@
 import pandas as pd
 from omegaconf import DictConfig
 
-from src.midst_toolkit.attacks.ensemble.utils import (
+from midst_toolkit.attacks.ensemble.data_utils import (
     save_dataframe,
 )
 
@@ -23,7 +23,7 @@ def expand_ranges(ranges):
     Returns:
         A flat list of integers covering the ranges.
     """
-    expanded = []
+    expanded: list = []
     for r in ranges:
         start, end = r
         expanded.extend(range(start, end))
@@ -165,7 +165,7 @@ def collect_population_data_ensemble(
         attack_types=challenge_attack_types,
         data_splits=["train", "dev", "final"],
         dataset="challenge",
-        data_cdata_processing_configonfig=data_processing_config,
+        data_processing_config=data_processing_config,
     )
     # Save the challenge points
     save_dataframe(df_challenge, save_dir, "challenge_points_all.csv")

examples/ensemble_attack_example/run_attack.py

@@ -7,15 +7,15 @@
 from pathlib import Path
 
 import hydra
+from ensemble_attack_example.real_data_collection import collect_population_data_ensemble
 from omegaconf import DictConfig
 
-from examples.ensemble_attack_example.real_data_collection import collect_population_data_ensemble
-from src.midst_toolkit.attacks.ensemble.process_split_data import process_split_data
-from src.midst_toolkit.common.logger import log
+from midst_toolkit.attacks.ensemble.process_split_data import process_split_data
+from midst_toolkit.common.logger import log
 
 
 @hydra.main(config_path=".", config_name="config", version_base=None)
-def main(cfg: DictConfig):
+def main(cfg: DictConfig) -> None:
     """
     Run the Ensemble Attack example pipeline.
     As the first step, data processing is done.

…/midst_toolkit/attacks/ensemble/utils.py …t_toolkit/attacks/ensemble/data_utils.pysrc/midst_toolkit/attacks/ensemble/utils.py renamed to src/midst_toolkit/attacks/ensemble/data_utils.py src/midst_toolkit/attacks/ensemble/utils.py renamed to src/midst_toolkit/attacks/ensemble/data_utils.py

@@ -3,7 +3,7 @@
 
 import pandas as pd
 
-from src.midst_toolkit.common.logger import log
+from midst_toolkit.common.logger import log
 
 
 def save_dataframe(df: pd.DataFrame, file_path: Path, file_name: str) -> None:

src/midst_toolkit/attacks/ensemble/process_split_data.py

@@ -5,10 +5,10 @@
 import pandas as pd
 from sklearn.model_selection import train_test_split
 
-from midst_toolkit.common.logger import log
-from src.midst_toolkit.attacks.ensemble.utils import (
+from midst_toolkit.attacks.ensemble.data_utils import (
     save_dataframe,
 )
+from midst_toolkit.common.logger import log
 
 
 def split_real_data(

tests/unit/attacks/ensemble/test_process_data_split.py

@@ -4,8 +4,8 @@
 from hydra import compose, initialize
 from omegaconf import DictConfig
 
-from src.midst_toolkit.attacks.ensemble.process_split_data import process_split_data
-from src.midst_toolkit.attacks.ensemble.utils import load_dataframe
+from midst_toolkit.attacks.ensemble.data_utils import load_dataframe
+from midst_toolkit.attacks.ensemble.process_split_data import process_split_data
 
 
 @pytest.fixture(scope="session")