Skip to content

Commit d6e007e

Browse files
committed
Fix no_polars test with missing 'without_import'
1 parent 7a214d5 commit d6e007e

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

test/test_stancsv.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -160,8 +160,9 @@ def test_csv_bytes_to_numpy_header_no_draws_no_polars():
160160
b"n_leapfrog__,divergent__,energy__,theta\n"
161161
),
162162
]
163-
arr = stancsv.csv_bytes_list_to_numpy(lines)
164-
assert arr.shape == (0, 8)
163+
with without_import("polars", cmdstanpy.utils.stancsv):
164+
arr = stancsv.csv_bytes_list_to_numpy(lines)
165+
assert arr.shape == (0, 8)
165166

166167

167168
def test_parse_comments_and_draws():

0 commit comments

Comments
 (0)