[FIX] status and set_type in CONFIG

NINAnor · Sep 5, 2023 · 6bf8c06 · 6bf8c06
1 parent bbabd8d
commit 6bf8c06
Showing 1 changed file with 2 additions and 18 deletions.
diff --git a/data_utils/DCASEfewshot.py b/data_utils/DCASEfewshot.py
@@ -543,22 +543,6 @@ def preprocess_df(df):
 if __name__ == "__main__":
     parser = argparse.ArgumentParser()
 
-    parser.add_argument(
-        "--status",
-        help=" 'train' or 'validate' or 'test'",
-        default="train",
-        required=False,
-        type=str,
-    )
-
-    parser.add_argument(
-        "--set_type",
-        help=" 'Training_Set' or 'Validation_Set' or 'Evaluation_Set",
-        default="Training_Set",
-        required=False,
-        type=str,
-    )
-
     parser.add_argument(
         "--overwrite",
         help="If there's an existing folder, should it be deleted?",
@@ -598,8 +582,8 @@ def preprocess_df(df):
         cfg = yaml.load(f, Loader=FullLoader)
 
     prepare_training_val_data(
-        cli_args.status,
-        cli_args.set_type,
+        cfg["data"]["status"],
+        cfg["data"]["set_type"],
         cli_args.overwrite,
         cfg["data"]["tensor_length"],
         cfg["data"]["frame_length"],