Skip to content

Commit 672cd4d

Browse files
committed
Add health check entrypoints to pyproject.toml
1 parent bc883a6 commit 672cd4d

1 file changed

Lines changed: 6 additions & 0 deletions

File tree

python/cuml/pyproject.toml

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -143,6 +143,12 @@ test-dask = [
143143
Homepage = "https://github.com/rapidsai/cuml"
144144
Documentation = "https://docs.rapids.ai/api/cuml/stable/"
145145

146+
[project.entry-points.rapids_doctor_check]
147+
cuml_import = "cuml.health_checks:import_check"
148+
cuml_functional = "cuml.health_checks:functional_check"
149+
cuml_accel_basic = "cuml.health_checks:accel_basic_check"
150+
cuml_accel_cli = "cuml.health_checks:accel_cli_check"
151+
146152
[tool.black]
147153
line-length = 79
148154
target-version = ["py311"]

0 commit comments

Comments
 (0)