@@ -646,19 +646,29 @@ def test_sample_no_params() -> None:
646646 stan = os .path .join (DATAFILES_PATH , 'datagen_poisson_glm.stan' )
647647 datagen_model = CmdStanModel (stan_file = stan )
648648 datagen_fit = datagen_model .sample (iter_sampling = 100 , show_progress = False )
649- assert np .isnan (datagen_fit .step_size ).all ()
650649 summary = datagen_fit .summary ()
651- assert 'lp__' in list (summary .index )
650+
651+ if cmdstan_version_before (2 , 36 ):
652+ assert 'lp__' not in list (summary .index )
653+ assert datagen_fit .step_size is None
654+ else :
655+ assert 'lp__' in list (summary .index )
656+ assert np .isnan (datagen_fit .step_size ).all ()
652657
653658 exe_only = os .path .join (DATAFILES_PATH , 'exe_only' )
654659 shutil .copyfile (datagen_model .exe_file , exe_only )
655660 os .chmod (exe_only , 0o755 )
656661 datagen2_model = CmdStanModel (exe_file = exe_only )
657662 datagen2_fit = datagen2_model .sample (iter_sampling = 200 , show_console = True )
658663 assert datagen2_fit .chains == 4
659- assert np .isnan (datagen2_fit .step_size ).all ()
660664 summary = datagen2_fit .summary ()
661- assert 'lp__' in list (summary .index )
665+
666+ if cmdstan_version_before (2 , 36 ):
667+ assert datagen2_fit .step_size is None
668+ assert 'lp__' not in list (summary .index )
669+ else :
670+ assert np .isnan (datagen2_fit .step_size ).all ()
671+ assert 'lp__' in list (summary .index )
662672
663673
664674def test_index_bounds_error () -> None :
0 commit comments