%g\ dZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl m Z ddlmZddlmZddlmZmZmZmZmZmZddlmZddlmZdd lmZmZm Z m!Z!m"Z"dd l#m$Z$d d l%m&Z'ejPd k\serddl)m*Z*n ddl)m+Z+dZ* ddl,Z,e,jZ-Gdde/Z0Gdde/Z1de2fdZ3ejhde3sdndZ5e3rdndZ6dee7e7ffdZ8de7fdZ9de7fdZ:Gdd Z;d!e7d"e2de2fd#Z<Gd$d%Z=d;d&e2ddfd'Z>dd1e7d(e2ddfd3ZGd4e7d&e2d(e2ddfd5ZHd6ee=e;fddfd7ZIdee7effd8ZJd?d9ZKeLd:k(reKyy#e.$rY)wxYw)@a Download and install a CmdStan release from GitHub. Downloads the release tar.gz file to temporary storage. Retries GitHub requests in order to allow for transient network outages. Builds CmdStan executables and tests the compiler by building example model ``bernoulli.stan``. Optional command line arguments: -i, --interactive: flag, when specified ignore other arguments and ask user for settings on STDIN -v, --version : version, defaults to latest release version -d, --dir : install directory, defaults to '$HOME/.cmdstan --overwrite: flag, when specified re-installs existing version --progress: flag, when specified show progress bar for CmdStan download --verbose: flag, when specified prints output from CmdStan build process --cores: int, number of cores to use when building, defaults to 1 -c, --compiler : flag, add C++ compiler to path (Windows only) N) OrderedDict)Path)sleep) TYPE_CHECKINGAnyCallableDictOptionalUnion)tqdm) _DOT_CMDSTAN) cmdstan_path do_commandpushd validate_dirwrap_url_progress_hook)get_download_url)progress))cached_property) lru_cachec8ttd|S)N)maxsize)propertyr)funs V/var/www/dash_apps/app1/venv/lib/python3.12/site-packages/cmdstanpy/install_cmdstan.pyrr9s/ $/455c eZdZy)CmdStanRetrieveErrorN__name__ __module__ __qualname__rrr!r!Hrr!c eZdZy)CmdStanInstallErrorNr"r&rrr)r)Lr'rr)returnc0tjdk(S)NWindows)platformsystemr&rr is_windowsr/Ps ??  ))rMAKEmakez mingw32-makez.execti}tjjd}|dj||d<|S)zCreate headers dictionary. GITHUB_PATztoken {} Authorization)osenvirongetformat)headersr4s r get_headersr;Xs9G -J#-#4#4Z#@ Nrcpd}tjj|t}t dD]1} tjj |j }ntjjd }|d }t!j"d |}||j%d}|S#tjj$rY}tdt||dkr-tdj|dztdYd}~td |d}~wwxYw) z&Report latest CmdStan release version.z=https://api.github.com/repos/stan-dev/cmdstan/releases/latest)r:zCannot connect to github. retry ({}/5)rNz&Cannot connect to CmdStan github repo.zutf-8tag_namezv?(.+))urllibrequestRequestr;rangeurlopenreaderrorURLErrorprintr9rr!jsonloadsdecoderesearchgroup)urlrBiresponseecontenttagmatchs rlatest_versionrWas  ICnn$$S+-$@G 1X  ~~--g6;;=H  jj12G * C IIi %E kk!n J||$$  - . !H1un++AE23a&8  s-B??D5AD0$ D00D5ctjjtjjdtS)N~)r6path expanduserjoinr r&rr home_cmdstanr]{s& 77  bggll3 = >>rcVeZdZdZdddddddddeedeeded ed ed ed ed efdZ y)InstallationSettingsz/ A static installation settings object NFr)versiondirrverbose overwritecorescompilerr`rarrbrcrdrekwargsc |r|n t|_|r|n t|_||_||_||_||_|xr t|_ |} y)N) rWr`r]rarrbrcrdr/re) selfr`rarrbrcrdrerf_s r__init__zInstallationSettings.__init__sN#*w~/? 3<>   "  1Z\ r) r#r$r%__doc__r strboolintrrjr&rrr_r_s{"&!#c]     rr_answerdefaultc:|j}|dvry|dvry|S)N)yyesT)nnoF)lower)rorps ryes_norws( \\^F   NrceZdZdZedefdZedefdZedefdZ edefdZ edefdZ edefdZ ede fd Zy ) InteractiveSettingsa) Installation settings provided on-demand in an interactive format. This provides the same set of properties as the ``InstallationSettings`` object, but rather than them being fixed by the constructor the user is asked for input whenever they are accessed for the first time. r*cjt}tdtd|td}|r|S|S)Nz(Which version would you like to install? Default: z'Type version or hit enter to continue: )rWrIinput)rhlatestros rr`zInteractiveSettings.versions:! 89 &"#@Av+V+rct}tdtd|td}|rtjj |S|S)Nz(Where would you like to install CmdStan?r{z)Type full path or hit enter to continue: )r]rIr|r6rZr[)rh directoryros rrazInteractiveSettings.dirsH N  89 )%&BC-3rww!!&)BBrc\tdtdtd}t|dS)Nz Show installation progress bars?z Default: y[y/n]: TrIr|rwrhros rrzInteractiveSettings.progresss* 01 ly!fd##rc\tdtdtd}t|dS)Nz0Show verbose output of the installation process? Default: nrFrrs rrbzInteractiveSettings.verboses* @A ly!fe$$rc\tdtdtd}t|dS)Nz(Overwrite existing CmdStan installation?rrFrrs rrczInteractiveSettings.overwrites* 89 ly!fe$$rctsytdtdtdtdtd}t|dS)NFz5Would you like to install the RTools40 C++ toolchain?z(A C++ toolchain is required for CmdStan.ztIf you are not sure if you need the toolchain or not, the most likely case is you do need it, and should answer 'y'.rr)r/rIr|rwrs rrezInteractiveSettings.compilersK| EF 89  M  ly!fe$$rctjxsd}tdtd|td} t |t t |dS#t$rYywxYw)NrzNHow many CPU cores would you like to use for installing and compiling CmdStan?zDefault: 1, Max: z)Enter a number or hit enter to continue: )r6 cpu_countrIr|minmaxrn ValueError)rhmax_cpusros rrdzInteractiveSettings.coressk<<>&Q  %  !(,-BC xS[!!45 5  sA A)(A)N)r#r$r%rkrrlr`rarmrrbrcrernrdr&rrryrys,,,CSCC$$$$ %%% %4%%  %$ % % s  rryrbctdg} |r t|yt|dy#t$r}tdt |d}~wwxYw)z Run `make clean-all` in the current directory (must be a cmdstan library). :param verbose: Boolean value; when ``True``, show output from make command. z clean-allNfd_outz Command "make clean-all" failed )r0r RuntimeErrorr)rl)rbcmdrSs r clean_allrsQ  CP  sO s4 ( P!$Ec!fX"NOOPs & & A AA rrdc tdd|g} |r t|n<|r-tjrt }t|d|n t|dtjjtjjddtzst d td tjjtjjdd tzst d td trtjjtjd d dd d}djtt!j"|gtj$j'ddj)dztj$d<yy#t $r}t dt|d}~wwxYw)a Run command ``make build`` in the current directory, which must be the home directory of a CmdStan version (or GitHub repo). By default, displays a progress bar which tracks make command outputs. If argument ``verbose=True``, instead of a progress bar, streams make command outputs to sys.stdout. When both ``verbose`` and ``progress`` are ``False``, runs silently. :param verbose: Boolean value; when ``True``, show output from make command. Default is ``False``. :param progress: Boolean value; when ``True`` display progress progress bar. Default is ``True``. :param cores: Integer, number of cores to use in the ``make`` command. Default is 1 core. buildz-jN)rpbarrzCommand "make build" failed bin stansummaryzbin/stansummaryz+ not found, please rebuild or report a bug!diagnosestanlib stan_mathtbb;PATHr2)r0rprogbarallow_show_progress_wrap_build_progress_hookrr)rlr6rZexistsr\ EXTENSIONr/getcwdlistrfromkeysr7r8split)rbrrdr progress_hookrSlibtbbs rrrs Bug, 'C L  sO '557!:!>"'',,umi.GH I!i[)0 0   77>>"'',,uj9.DE F!i[)0 0  | IIK UE !XX $$Hrzz~~fb9??DD   6  L!$A#a&"JKKLsA F77 GGGc^d}dtddd|igddd d td d ffd }|S)z7Sets up tqdm callback for CmdStan sampler console msgs.z z0{desc} ({elapsed}) | {bar} | {postfix[0][value]}valuezBuilding CmdStan bluer2r)total bar_formatpostfixcolourdescpositionliner*Nc|jdrRjd|jdd<jjz j y|jdr|jdd<y|ddd|ddjdd<jd jd y) Nz --- CmdStanDonerrz--rz ... i Compilingr) startswithset_descriptionrupdatertclose)r msgs_expectedrs rbuild_progress_hookz6_wrap_build_progress_hook..build_progress_hookGs ??= )   ('+DLLOG $ KK . / JJLt$+/ Q(.22AhZuT#$ZL+I Q($$[1 Ar)r rl)padrrrs @@rrr9sV CME.se456  D # $  rctdddjt}|jr|j t |j g} |r t|n t|d|js tdy#t$r%}tddj|d|d}~wwxYw) z Compile the example model. The current directory must be a cmdstan installation, i.e., contains the makefile, Stanc compiler, and all libraries. :param verbose: Boolean value; when ``True``, show output from make command. examples bernoulliNrz Command " z " failed: z!Failed to generate example binary) r with_suffixris_fileunlinkr0as_posixrrr)r\)rbrZrrSs rcompile_examplerXs  K 5 A A) LD ||~  !CM  sO s4 ( <<>!"EFF  M!Ichhsm_Ks"KLLMsB B= B88B=c tt5t|t|||t |dddy#1swYyxYw#t $r}t d|d}~wwxYw)a Rebuilds the existing CmdStan installation. This assumes CmdStan has already been installed, though it need not be installed via CmdStanPy for this function to work. :param verbose: Boolean value; when ``True``, show output from make command. Default is ``False``. :param progress: Boolean value; when ``True`` display progress progress bar. Default is ``True``. :param cores: Integer, number of cores to use in the ``make`` command. Default is 1 core. Nz8Failed to rebuild CmdStan. Are you sure it is installed?)rrrrrrr))rbrrdrSs rrebuild_cmdstanrrsf  <> " % g  '8U + G $ % % % ! F  s3A$AAA A A A( A##A(cmdstan_versionrcct|5tdj||r^tjj dr?tdj|t |tdj|t|||dddtdj|y#1swY$xYw)a Build specified CmdStan version by spawning subprocesses to run the Make utility on the downloaded CmdStan release src files. Assumes that current working directory is parent of release dir. :param cmdstan_version: CmdStan release, corresponds to release dirname. :param overwrite: when ``True``, run ``make clean-all`` before building. :param verbose: Boolean value; when ``True``, show output from make command. zHBuilding version {}, may take several minutes, depending on your system..z8Overwrite requested, remove existing build of version {}zRebuilding version {})rrdNz Installed {})rrIr9r6rZrrrrrcrbrrds rinstall_versionrs   7  ((.(?  , VO,  g  )00A B g6 7 .   01 7 7s B B88Cr`c ld|vryd}t|}tdD]"} tjj |$|S#tj j $r7}td|d|td|jd}Yd}~|Sd}~wtj j$r}|dkr.td j||d ztd Yd}~td j||td j|jd}Yd}~d}~wwxYw) Ngit:Tr=zRelease z is unavailable from URL z HTTPError: Fr>z.checking version {} availability, retry ({}/5)rz%Release {} is unavailable from URL {}z URLError: {}) rrDrArBrErG HTTPErrorrIcoderHr9rreason)r` is_availablerPrQerrrSs ris_version_availablers L 7 #C 1X! ! NN " "3 '!( #||%%  HWI%>seD E Kz* + L  ||$$ !1uDKKQ a 9@@#N O .''1 2 L !s*AD3"+B D34.D.'AD..D3c ||dk(r tdd|vr^|jdd}|jddjdd }td |d t d d ddd|dddd|g ytdj |t |}tdD]L} |rtjr t}nd}tjj|d|\}}ntdj  tdt)j*|} | j-} | | j.} nd} d|} | | k7rt1dj | | t3j4}t7rd j |}|r`tjrLt9| j;t=| j;d!d"#D]}| j?|$n| jA| jGtd'| y#tjj$r+} t!dj | j"|| d} ~ wtjj$$r} tdj |t| |dkr.tdj |dzt'dYd} ~ Ktdj |t!dj || d} ~ wwxYw#tB$r } t1d%|d&tE| | d} ~ wwxYw# jGwxYw)(z#Download specified CmdStan version.Nr2zArgument "version" unspecified.r:r-/rizCloning CmdStan branch 'z!' from stan-dev/cmdstan on GitHubgitclonez--depth1z--branchz --recursivez--shallow-submodulesz'https://github.com/stan-dev/cmdstan.gitcmdstan-zDownloading CmdStan version {}r=)filename reporthookz7HTTPError: {} Version {} not available from github.com.z5Failed to download CmdStan version {} from github.comr>r?z)Version {} not available from github.com.zDownload successful, file: {}zExtracting distributionzAtarfile should contain top-level dir {},but found dir {} instead.z\\?\{}rF)iterablerrleave)memberzFailed to unpack file z , error: zUnpacked download as )$rrreplacerIrr9rrDrrrrArB urlretrieverGrr!rrHrtarfileopennextnamer)r6rr/r getmemberslenextract extractall Exceptionrlr)r`rrU tag_folderrPrQrfile_tmprirStarfirsttop_dir cmdstan_dirtargetrs rretrieve_versionrsT'R-:;; mmC #__S#.66sC@  (-NOP&9:,'    * 1 1' :; 7 #C 1X! G779+,!%  ..44d}5KHa !D ) 0 0 :;" '(ll8$   jjGG  * k !%,,2F;,H  <%%f-F 335)#..*+  +  6 *  + NN   !+ /0u||%% &<$SE*$T\\N3 txx *GGNN?;  GGLLi'  A !DNN#DLL16==dllKL t||n-MM MM/ > T\\4== 9  /+> jj   8?? M N ? # * * + DLL ) *= * *< * *= * *s%-E,J!J :J J JJ ctjd}|jdddd|jddd |jd d d |jddd|jddd|jddd|jddtdt r|jdddddt |j tjddS)Ninstall_cmdstanz --interactivez-i store_truezCIgnore other arguments and run the installation in interactive mode)actionhelpz --versionz-vz|version, defaults to latest release version. If git is installed, you can also specify a git tag or branch, e.g. git:develop)rz--dirz-dz.install directory, defaults to '$HOME/.cmdstanz --overwritez1flag, when specified re-installs existing versionz --verbosez=flag, when specified prints output from CmdStan build processz --progressz;flag, when specified show progress bar for CmdStan downloadz--coresrz%number of cores to use while building)rptyperz --compilerz-crez-flag, add C++ compiler to path (Windows only))destrr) argparseArgumentParser add_argumentrnr/vars parse_argssysargv)parsers rparse_cmdline_argsr#s9  $ $%6 7F      L  @   L   J    4  |   @   !!#((12,/ 00rct}|jddrttytt di|y)N interactiveFr&)r#r8rryr_)r s r__main__r&s5  D xx u%')*(0401rr&)F)FTr)FFTr)T)r*N)MrkrrJr6r-rMrr r urllib.errorrAurllib.request collectionsrpathlibrtimertypingrrrr r r tqdm.autor cmdstanpyr cmdstanpy.utilsrrrrrcmdstanpy.utils.cmdstanrr2rr version_info functoolsrrreadlineri ImportErrorrr!r)rmr/getenvr0rrlr;rWr]r_rwryrrnr wrap_callbackrrrrrrr rr#r&r#r&rrr7s$  #FF"5!v)$6  A  <  , *D*ryyz|H lF T#s(^4?c? :3$HHVPtPP$3 43 43 s3 43 l 8HcUD[,A#B<GTGdG6@A %)9< : 2222 2  2  2@#$:a1ca1Ta1Ta1H!)c!)D!)D!)T!)H3*e/1EEF3*43*l31DcN31l2 z JE  sGGG