Remove useless imports

takuseno · Nov 8, 2024 · eb7ee38 · eb7ee38
1 parent 6648645
commit eb7ee38
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 4 deletions.
diff --git 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
@@ -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
@@ -43,7 +43,7 @@
             "colorama",
             "dataclasses-json",
             "gymnasium>=1.0.0",
-            "scipy",
+            "scikit-learn",
         ],
         packages=find_packages(exclude=["tests*"]),
         python_requires=">=3.9.0",