fixes

WIMU-BKT · Jan 13, 2025 · 92140af · 92140af
1 parent a941f04
commit 92140af
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 4 deletions.
diff --git a/wimudp/data_poisoning/generate_dirty_label.py b/wimudp/data_poisoning/generate_dirty_label.py
@@ -14,7 +14,7 @@ def main():
     samples_number = int(samples_number) if samples_number.isdigit() else None
 
     display_stage("Filtering data")
-    df_filtered = process_csv_file(concept_a=concept_a or None, concept_a_action=concept_a_action or None, rows_number=rows_number)
+    df_filtered = process_csv_file(concept_a=concept_a or None, concept_c_action=concept_a_action or None, rows_number=rows_number)
     print("Filtered DataFrame:")
     print(df_filtered)
 

diff --git a/wimudp/data_poisoning/generate_nightshade.py b/wimudp/data_poisoning/generate_nightshade.py
@@ -34,7 +34,5 @@ def display_stage(stage_name):
     print(f"{' ' * ((50 - len(stage_name)) // 2)}{stage_name.upper()}")
     print("=" * 50 + "\n")
 
-display_stage("Filtering Data")
-
 if __name__ == "__main__":
     main()
diff --git a/wimudp/data_poisoning/utils.py b/wimudp/data_poisoning/utils.py
@@ -4,7 +4,7 @@
 
 from typing import Tuple
 
-AUDIOLDM_DATASET_DIR = "../finetuning/audioldm/data/dataset"
+AUDIOLDM_DATASET_DIR = "./finetuning/audioldm/data/dataset"
 AUDIOS_DIR = f"{AUDIOLDM_DATASET_DIR}/audioset"
 CONCEPT_A = "dog"
 CONCEPT_C = "cat"