diff --git a/README.md b/README.md index 868a781a..ce8a1a8e 100644 --- a/README.md +++ b/README.md @@ -101,7 +101,7 @@ $ docker run -it --gpus all --name d3rlpy takuseno/d3rlpy:latest bash | [Critic Reguralized Regression (CRR)](https://arxiv.org/abs/2006.15134) | :no_entry: | :white_check_mark: | | [Policy in Latent Action Space (PLAS)](https://arxiv.org/abs/2011.07213) | :no_entry: | :white_check_mark: | | [TD3+BC](https://arxiv.org/abs/2106.06860) | :no_entry: | :white_check_mark: | -| [PRDC](https://arxiv.org/abs/2306.06569) | :no_entry: | :white_check_mark: | +| [Policy Regularization with Dataset Constraint (PRDC)](https://arxiv.org/abs/2306.06569) | :no_entry: | :white_check_mark: | | [Implicit Q-Learning (IQL)](https://arxiv.org/abs/2110.06169) | :no_entry: | :white_check_mark: | | [Calibrated Q-Learning (Cal-QL)](https://arxiv.org/abs/2303.05479) | :no_entry: | :white_check_mark: | | [ReBRAC](https://arxiv.org/abs/2305.09836) | :no_entry: | :white_check_mark: | diff --git a/d3rlpy/algos/qlearning/torch/prdc_impl.py b/d3rlpy/algos/qlearning/torch/prdc_impl.py index 79e5fd6f..4add2280 100644 --- a/d3rlpy/algos/qlearning/torch/prdc_impl.py +++ b/d3rlpy/algos/qlearning/torch/prdc_impl.py @@ -1,9 +1,7 @@ # pylint: disable=too-many-ancestors import dataclasses -import numpy as np import torch -from scipy.spatial import KDTree from sklearn.neighbors import NearestNeighbors from ....models.torch import ActionOutput, ContinuousEnsembleQFunctionForwarder diff --git a/setup.py b/setup.py index 5ff9eb9b..0505e347 100644 --- a/setup.py +++ b/setup.py @@ -43,7 +43,7 @@ "colorama", "dataclasses-json", "gymnasium>=1.0.0", - "scipy", + "scikit-learn", ], packages=find_packages(exclude=["tests*"]), python_requires=">=3.9.0",