hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-08 11:31:40 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

Enable test_joblib in CI (#7404)

Browse source
This commit is contained in:
Edward Oakes 2020-03-02 20:03:27 -06:00 • committed by GitHub
parent d69fe54f6d
commit 2b6f00724a
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
python/ray/tests/test_joblib.py
View file

@ -1,8 +1,11 @@
import numpy as np
import joblib import joblib
import sys
import time
import numpy as np
from sklearn.datasets import load_digits, load_iris from sklearn.datasets import load_digits, load_iris
from sklearn.model_selection import RandomizedSearchCV from sklearn.model_selection import RandomizedSearchCV
from time import time
from sklearn.datasets import fetch_openml from sklearn.datasets import fetch_openml
from sklearn.ensemble import ExtraTreesClassifier from sklearn.ensemble import ExtraTreesClassifier
from sklearn.ensemble import RandomForestClassifier from sklearn.ensemble import RandomForestClassifier
@ -142,9 +145,9 @@ def test_sklearn_benchmarks(ray_start_cluster_2_nodes):
if "n_jobs" in estimator_params: if "n_jobs" in estimator_params:
estimator.set_params(n_jobs=num_jobs) estimator.set_params(n_jobs=num_jobs)
time_start = time() time_start = time.time()
estimator.fit(X_train, y_train) estimator.fit(X_train, y_train)
train_time[name] = time() - time_start train_time[name] = time.time() - time_start
print("training", name, "took", train_time[name], "seconds") print("training", name, "took", train_time[name], "seconds")
@ -157,3 +160,8 @@ def test_cross_validation(shutdown_only):
assert len(accuracy) == 5 assert len(accuracy) == 5
for result in accuracy: for result in accuracy:
assert result > 0.95 assert result > 0.95
if __name__ == "__main__":
import pytest
sys.exit(pytest.main(["-v", __file__]))