Skip to content

Commit 25d3edb

Browse files
committed
bug: fix adding package files to wheel
1 parent 9d3347b commit 25d3edb

2 files changed

Lines changed: 13 additions & 6 deletions

File tree

hatch_build.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -33,11 +33,8 @@ def initialize(self, version: str, build_data: dict) -> None: # noqa: ARG002
3333
"""
3434
# 1. Validate environment
3535
if not os.environ.get("GCP_PROJECT"):
36-
msg = (
37-
"GCP_PROJECT environment variable is not set. "
38-
"This is required to generate index data files."
39-
)
40-
raise OSError(msg)
36+
self.app.display_warning("Skipping data generation: GCP_PROJECT not set")
37+
return
4138

4239
# 2. Read build options from environment or config
4340
# Allow override via env vars, matching CMake behavior
@@ -62,7 +59,9 @@ def initialize(self, version: str, build_data: dict) -> None: # noqa: ARG002
6259
Path(__file__).parent / "scripts" / "python" / "idc_index_data_manager.py"
6360
)
6461

65-
cmd = [sys.executable, str(script_path)]
62+
root_path = Path(self.root)
63+
64+
cmd = [sys.executable, str(script_path), "--output-dir", str(root_path)]
6665
if generate_csv:
6766
cmd.append("--generate-csv-archive")
6867
if generate_parquet:
@@ -98,6 +97,7 @@ def initialize(self, version: str, build_data: dict) -> None: # noqa: ARG002
9897
self.app.display_info(f"Registered: {csv_file.name}")
9998

10099
if generate_parquet:
100+
# Register parquet files (only main indices to reduce package size)
101101
parquet_files = [
102102
"idc_index.parquet",
103103
"prior_versions_index.parquet",

scripts/python/idc_index_data_manager.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -463,6 +463,12 @@ def retrieve_latest_idc_release_version(self) -> int:
463463
action="store_true",
464464
help="Generate idc_index.parquet file",
465465
)
466+
parser.add_argument(
467+
"--output-dir",
468+
type=Path,
469+
default=None,
470+
help="Output directory for generated files (default: current directory)",
471+
)
466472
parser.add_argument(
467473
"--retrieve-latest-idc-release-version",
468474
action="store_true",
@@ -480,6 +486,7 @@ def retrieve_latest_idc_release_version(self) -> int:
480486
IDCIndexDataManager(args.project).generate_index_data_files(
481487
generate_compressed_csv=args.generate_csv_archive,
482488
generate_parquet=args.generate_parquet,
489+
output_dir=args.output_dir,
483490
)
484491
elif args.retrieve_latest_idc_release_version:
485492
logging.basicConfig(level=logging.ERROR, force=True)

0 commit comments

Comments
 (0)