Skip to content

Commit

Permalink
update dataset install message
Browse files Browse the repository at this point in the history
  • Loading branch information
lobis committed Dec 8, 2023
1 parent b2f29d7 commit 1c6ea3a
Showing 1 changed file with 5 additions and 2 deletions.
7 changes: 5 additions & 2 deletions src/geant4_python_application/datasets.py
Original file line number Diff line number Diff line change
Expand Up @@ -173,7 +173,10 @@ def install_datasets(force: bool = False, show_progress: bool = True):
os.makedirs(data_dir, exist_ok=True)
if show_progress:
print(
f"""The following Geant4 datasets will be installed to {data_dir}: {", ".join([f"{dataset.name}@{dataset.version}" for dataset in datasets_to_download])}"""
f"Geant4 datasets (< 2GB) will be installed to temporary directory {data_dir}. This may take a while but only needs to be done once."
)
print(
f"""The following Geant4 datasets will be installed: {", ".join([f"{dataset.name}@{dataset.version}" for dataset in datasets_to_download])}"""
)

with tqdm(
Expand All @@ -196,7 +199,7 @@ def install_datasets(force: bool = False, show_progress: bool = True):
total_size_gb = sum(fp.stat().st_size for fp in Path(data_dir).rglob("*")) / (
1024**3
)
print(f"Geant4 datasets size on disk after extraction: {total_size_gb:.2f} GB")
print(f"Geant4 datasets size on disk after extraction: {total_size_gb:.2f}GB")


def uninstall_datasets():
Expand Down

0 comments on commit 1c6ea3a

Please sign in to comment.