fix: delete deplicate --output param

ChenZiHong-Gavin · ChenZiHong-Gavin · commit d60238e79dc0 · 2025-12-17T17:01:02.000+08:00
diff --git a/examples/evaluate/evaluate.sh b/examples/evaluate/evaluate.sh
@@ -1,4 +1,3 @@
 python3 -m graphgen.evaluate --folder cache/data \
-                    --output cache/output \
                     --reward "OpenAssistant/reward-model-deberta-v3-large-v2,BAAI/IndustryCorpus2_DataRater" \
                     --uni MingZhong/unieval-sum \
diff --git a/examples/extract/extract_schema_guided/extract_schema_guided.sh b/examples/extract/extract_schema_guided/extract_schema_guided.sh
@@ -1,3 +1,2 @@
 python3 -m graphgen.run \
---config_file examples/extract/extract_schema_guided/schema_guided_extraction_config.yaml \
---output_dir cache/
+--config_file examples/extract/extract_schema_guided/schema_guided_extraction_config.yaml
diff --git a/examples/generate/generate_aggregated_qa/generate_aggregated.sh b/examples/generate/generate_aggregated_qa/generate_aggregated.sh
@@ -1,3 +1,2 @@
 python3 -m graphgen.run \
---config_file examples/generate/generate_aggregated_qa/aggregated_config.yaml \
---output_dir cache/
+--config_file examples/generate/generate_aggregated_qa/aggregated_config.yaml
diff --git a/examples/generate/generate_atomic_qa/generate_atomic.sh b/examples/generate/generate_atomic_qa/generate_atomic.sh
@@ -1,3 +1,2 @@
 python3 -m graphgen.run \
---config_file examples/generate/generate_atomic_qa/atomic_config.yaml \
---output_dir cache/
+--config_file examples/generate/generate_atomic_qa/atomic_config.yaml
diff --git a/examples/generate/generate_cot_qa/generate_cot.sh b/examples/generate/generate_cot_qa/generate_cot.sh
@@ -1,3 +1,2 @@
 python3 -m graphgen.run \
---config_file examples/generate/generate_cot_qa/cot_config.yaml \
---output_dir cache/
+--config_file examples/generate/generate_cot_qa/cot_config.yaml
diff --git a/examples/generate/generate_multi_hop_qa/generate_multi_hop.sh b/examples/generate/generate_multi_hop_qa/generate_multi_hop.sh
@@ -1,3 +1,2 @@
 python3 -m graphgen.run \
---config_file examples/generate/generate_multi_hop_qa/multi_hop_config.yaml \
---output_dir cache/
+--config_file examples/generate/generate_multi_hop_qa/multi_hop_config.yaml
diff --git a/examples/generate/generate_vqa/generate_vqa.sh b/examples/generate/generate_vqa/generate_vqa.sh
@@ -1,3 +1,2 @@
 python3 -m graphgen.run \
---config_file examples/generate/generate_vqa/vqa_config.yaml \
---output_dir cache/
+--config_file examples/generate/generate_vqa/vqa_config.yaml
diff --git a/examples/search/search_dna.sh b/examples/search/search_dna.sh
@@ -1,4 +1,2 @@
 python3 -m graphgen.run \
---config_file graphgen/configs/search_dna_config.yaml \
---output_dir cache/
-
+--config_file graphgen/configs/search_dna_config.yaml
diff --git a/examples/search/search_rna.sh b/examples/search/search_rna.sh
@@ -1,4 +1,2 @@
 python3 -m graphgen.run \
---config_file graphgen/configs/search_rna_config.yaml \
---output_dir cache/
-
+--config_file graphgen/configs/search_rna_config.yaml
diff --git a/examples/search/search_uniprot.sh b/examples/search/search_uniprot.sh
@@ -1,3 +1,2 @@
 python3 -m graphgen.run \
---config_file graphgen/configs/search_protein_config.yaml \
---output_dir cache/
+--config_file graphgen/configs/search_protein_config.yaml
diff --git a/graphgen/run.py b/graphgen/run.py
@@ -66,21 +66,13 @@ def main():
         .joinpath("aggregated_config.yaml"),
         type=str,
     )
-    parser.add_argument(
-        "--output_dir",
-        help="Output directory for GraphGen.",
-        default=sys_path,
-        required=True,
-        type=str,
-    )
 
     args = parser.parse_args()
 
-    working_dir = args.output_dir
-
     with open(args.config_file, "r", encoding="utf-8") as f:
         config = yaml.load(f, Loader=yaml.FullLoader)
 
+    working_dir = config.get("global_params", {}).get("working_dir", "cache")
     unique_id = int(time.time())
     output_path = os.path.join(working_dir, "output", f"{unique_id}")
     set_working_dir(output_path)
