@@ -160,7 +160,7 @@ def test_download_prediction_csv_class(driver, project, dataset, featureset,
160
160
driver .get ('/' )
161
161
_click_download (project .id , driver )
162
162
matching_downloads_paths = glob .glob (f'{ cfg ["paths:downloads_folder" ]} /'
163
- 'cesium_prediction_results*.csv ' )
163
+ 'cesium_prediction_results*' )
164
164
assert len (matching_downloads_paths ) == 1
165
165
try :
166
166
npt .assert_equal (
@@ -180,7 +180,7 @@ def test_download_prediction_csv_class_unlabeled(driver, project, unlabeled_pred
180
180
driver .get ('/' )
181
181
_click_download (project .id , driver )
182
182
matching_downloads_paths = glob .glob (f'{ cfg ["paths:downloads_folder" ]} /'
183
- 'cesium_prediction_results*.csv ' )
183
+ 'cesium_prediction_results*' )
184
184
assert len (matching_downloads_paths ) == 1
185
185
try :
186
186
result = np .genfromtxt (matching_downloads_paths [0 ], dtype = 'str' )
@@ -196,7 +196,7 @@ def test_download_prediction_csv_class_prob(driver, project, dataset,
196
196
driver .get ('/' )
197
197
_click_download (project .id , driver )
198
198
matching_downloads_paths = glob .glob (f'{ cfg ["paths:downloads_folder" ]} /'
199
- 'cesium_prediction_results*.csv ' )
199
+ 'cesium_prediction_results*' )
200
200
assert len (matching_downloads_paths ) == 1
201
201
try :
202
202
result = pd .read_csv (matching_downloads_paths [0 ])
@@ -218,7 +218,7 @@ def test_download_prediction_csv_regr(driver, project, dataset, featureset,
218
218
driver .get ('/' )
219
219
_click_download (project .id , driver )
220
220
matching_downloads_paths = glob .glob (f'{ cfg ["paths:downloads_folder" ]} /'
221
- 'cesium_prediction_results*.csv ' )
221
+ 'cesium_prediction_results*' )
222
222
assert len (matching_downloads_paths ) == 1
223
223
try :
224
224
results = np .genfromtxt (matching_downloads_paths [0 ],
0 commit comments