Lint.

dmlc · Jul 15, 2020 · 74205b6 · 74205b6
1 parent ea0d4b1
commit 74205b6
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/python-package/xgboost/data.py b/python-package/xgboost/data.py
@@ -725,6 +725,8 @@ def _device_quantile_transform(data, feature_names, feature_types):
         return data, feature_names, feature_types
     if _is_dlpack(data):
         return _transform_dlpack(data), feature_names, feature_types
+    raise TypeError('Value type is not supported for data iterator:' +
+                    str(type(data)))
 
 
 def dispatch_device_quantile_dmatrix_set_data(proxy, data):

diff --git a/tests/python/test_basic.py b/tests/python/test_basic.py
@@ -68,7 +68,7 @@ def test_basic(self):
             # save dmatrix into binary buffer
             dtest.save_binary(dtest_path)
             # save model
-            model_path = 'model.booster'
+            model_path = os.path.join(tmpdir, 'model.booster')
             bst.save_model(model_path)
             # load model and data in
             bst2 = xgb.Booster(model_file=model_path)