Skip to content

Commit 356ecb4

Browse files
committed
ENH: package all small indices to simplify usage
1 parent 92cc40a commit 356ecb4

1 file changed

Lines changed: 6 additions & 4 deletions

File tree

hatch_build.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -16,9 +16,11 @@ class IDCBuildHook(BuildHookInterface):
1616

1717
PLUGIN_NAME = "custom"
1818

19-
PARQUET_WHITELIST: ClassVar[set[str]] = {
20-
"idc_index.parquet",
21-
"prior_versions_index.parquet",
19+
# Parquet files to exclude from the package to reduce size
20+
PARQUET_EXCLUDE_LIST: ClassVar[set[str]] = {
21+
"sm_index.parquet",
22+
"sm_instance_index.parquet",
23+
"clinical_index.parquet",
2224
}
2325

2426
def _prune_unwhitelisted_parquet_files(self) -> None:
@@ -28,7 +30,7 @@ def _prune_unwhitelisted_parquet_files(self) -> None:
2830
return
2931

3032
for parquet_file in package_dir.glob("*.parquet"):
31-
if parquet_file.name not in self.PARQUET_WHITELIST:
33+
if parquet_file.name in self.PARQUET_EXCLUDE_LIST:
3234
parquet_file.unlink()
3335
self.app.display_info(
3436
f"Removed non-whitelisted parquet: {parquet_file.name}"

0 commit comments

Comments
 (0)