File tree Expand file tree Collapse file tree 2 files changed +3
-3
lines changed
Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -74,7 +74,7 @@ def from_csv(
7474 elif isinstance (path , (str , os .PathLike )):
7575 if os .path .exists (path ) and os .path .isdir (path ):
7676 for file in os .listdir (path ):
77- if os .path .splitext (path )[1 ] == ".csv" :
77+ if os .path .splitext (file )[1 ] == ".csv" :
7878 csvfiles .append (os .path .join (path , file ))
7979 elif os .path .exists (path ):
8080 csvfiles .append (str (path ))
Original file line number Diff line number Diff line change @@ -50,7 +50,7 @@ def test_optimize_copy_ability(self):
5050
5151class VariationalCompliance (unittest .TestCase ):
5252 def test_variational_pickle_ability (self ):
53- csvfiles_path = DATAFILES_PATH / 'variational' / 'eta_big_output.csv'
53+ csvfiles_path = DATAFILES_PATH / 'variational'
5454 fit = cmdstanpy .from_csv (path = csvfiles_path )
5555 keys = fit .stan_variables ().keys ()
5656 pickled = pickle .dumps (fit )
@@ -59,7 +59,7 @@ def test_variational_pickle_ability(self):
5959 self .assertSequenceEqual (keys , unpickled .stan_variables ().keys ())
6060
6161 def test_variational_copy_ability (self ):
62- csvfiles_path = DATAFILES_PATH / 'variational' / 'eta_big_output.csv'
62+ csvfiles_path = DATAFILES_PATH / 'variational'
6363 fit = cmdstanpy .from_csv (path = csvfiles_path )
6464 fit2 = copy .deepcopy (fit )
6565 self .assertSequenceEqual (
You can’t perform that action at this time.
0 commit comments