diff --git a/pyproject.toml b/pyproject.toml index f231c59..c4f779c 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -29,8 +29,8 @@ optuna = {version = ">=2.10", optional = true} attrs = {version = ">=22.1.0", optional = true} [tool.poetry.extras] -_torch_cpu = ["torch"] -_tensorflow = ["tensorflow-cpu"] +x-torch_cpu = ["torch"] +x-tensorflow = ["tensorflow-cpu"] optuna = ["optuna"] attrs = ["attrs"]