Spaces:
Running
Running
github-actions[bot]
commited on
Commit
·
8dd5b0a
1
Parent(s):
982cb95
Auto-sync from demo at Wed Dec 17 09:02:01 UTC 2025
Browse files- graphgen/run.py +1 -9
graphgen/run.py
CHANGED
|
@@ -66,21 +66,13 @@ def main():
|
|
| 66 |
.joinpath("aggregated_config.yaml"),
|
| 67 |
type=str,
|
| 68 |
)
|
| 69 |
-
parser.add_argument(
|
| 70 |
-
"--output_dir",
|
| 71 |
-
help="Output directory for GraphGen.",
|
| 72 |
-
default=sys_path,
|
| 73 |
-
required=True,
|
| 74 |
-
type=str,
|
| 75 |
-
)
|
| 76 |
|
| 77 |
args = parser.parse_args()
|
| 78 |
|
| 79 |
-
working_dir = args.output_dir
|
| 80 |
-
|
| 81 |
with open(args.config_file, "r", encoding="utf-8") as f:
|
| 82 |
config = yaml.load(f, Loader=yaml.FullLoader)
|
| 83 |
|
|
|
|
| 84 |
unique_id = int(time.time())
|
| 85 |
output_path = os.path.join(working_dir, "output", f"{unique_id}")
|
| 86 |
set_working_dir(output_path)
|
|
|
|
| 66 |
.joinpath("aggregated_config.yaml"),
|
| 67 |
type=str,
|
| 68 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 69 |
|
| 70 |
args = parser.parse_args()
|
| 71 |
|
|
|
|
|
|
|
| 72 |
with open(args.config_file, "r", encoding="utf-8") as f:
|
| 73 |
config = yaml.load(f, Loader=yaml.FullLoader)
|
| 74 |
|
| 75 |
+
working_dir = config.get("global_params", {}).get("working_dir", "cache")
|
| 76 |
unique_id = int(time.time())
|
| 77 |
output_path = os.path.join(working_dir, "output", f"{unique_id}")
|
| 78 |
set_working_dir(output_path)
|