Skip to content

Commit e377a1e

Browse files
fix: fix config path when packaging
1 parent d34e0a5 commit e377a1e

File tree

2 files changed

+5
-1
lines changed

2 files changed

+5
-1
lines changed

MANIFEST.in

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,2 @@
1+
include graphgen/configs/*.yaml
2+
recursive-include graphgen/configs *.yaml

graphgen/generate.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
import json
33
import time
44
import argparse
5+
from importlib.resources import files
56
import yaml
67
from dotenv import load_dotenv
78

@@ -28,7 +29,8 @@ def main():
2829
parser = argparse.ArgumentParser()
2930
parser.add_argument('--config_file',
3031
help='Config parameters for GraphGen.',
31-
default=os.path.join(sys_path, "configs", "graphgen_config.yaml"),
32+
# default=os.path.join(sys_path, "configs", "graphgen_config.yaml"),
33+
default=files('graphgen').joinpath("configs", "graphgen_config.yaml"),
3234
type=str)
3335
parser.add_argument('--output_dir',
3436
help='Output directory for GraphGen.',

0 commit comments

Comments
 (0)