saymrwulf/prophet
1
0
Fork 0
mirror of https://github.com/saymrwulf/prophet.git synced 2026-05-14 20:48:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix setup.py stan file temp destination (#2451)

Browse source
This commit is contained in:
Cuong Duong 2023-06-19 07:39:18 +10:00 committed by GitHub
parent 9aa3f7b0fd
commit 3746765007
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
python/setup.py
View file

@ -95,7 +95,6 @@ def install_cmdstan_deps(cmdstan_dir: Path):
cores=cpu_count(),
progress=True,
):
raise RuntimeError("CmdStan failed to install in repackaged directory")
@ -121,13 +120,13 @@ def build_cmdstan_model(target_dir):
install_cmdstan_deps(cmdstan_dir)
model_name = "prophet.stan"
temp_stan_file = copy(os.path.join(MODEL_DIR, model_name), cmdstan_dir)
# note: ensure copy target is a directory not a file.
temp_stan_file = copy(os.path.join(MODEL_DIR, model_name), cmdstan_dir.parent.resolve())
sm = cmdstanpy.CmdStanModel(stan_file=temp_stan_file)
target_name = "prophet_model.bin"
copy(sm.exe_file, os.path.join(target_dir, target_name))
if IS_WINDOWS:
if IS_WINDOWS and repackage_cmdstan():
copytree(cmdstan_dir, target_cmdstan_dir)
# Clean up