remove unused libraries (#454)

* remove numba

* remove pyvirtualdisplay and scikit dependencies

docs/installation.md

@@ -19,7 +19,7 @@ $ pip install rlberry
 To install rlberry with more options, you can use ``pip install rlberry[xxxxxxxx]``, with `xxxxxxxx` as :
 
 - `torch` to install `opencv-python, ale-py, stable-baselines3, tensorboard, torch`
-- `extras` to install `optuna, numba, ffmpeg-python, pyvirtualdisplay, scikit-fda, scikit-learn`
+- `extras` to install `optuna, ffmpeg-python, scikit-fda`
 
 (for dev)
 

pyproject.toml

@@ -33,16 +33,13 @@ moviepy = "*"
 pyopengl = "*"
 optuna ={version="*", optional=true}
 ffmpeg-python = {version="*", optional=true}
-pyvirtualdisplay = {version="*", optional=true}
 opencv-python = {version="*", optional=true}
 ale-py = {version="*", optional=true}
 stable-baselines3 = {version="*", optional=true}
 tensorboard = {version="*", optional=true}
 torch = {version=">=2.0.0, !=2.0.1, !=2.1.0", optional=true}
 pandas = "*"
-numba = "^0.58.1"
 scikit-fda = {version="^0.9", optional=true}
-scikit-learn = {version="1.2.2", optional=true}
 sphinx = {version="6.2.1", optional=true}
 sphinx-gallery = { version= "^0.14.0", optional=true}
 sphinx-math-dollar = {version="^1.2.1", optional=true}
@@ -60,7 +57,7 @@ multimethod = {version="1.10", optional=true}     # new version 1.11 : crash our
 
 [tool.poetry.extras]
 torch = ["opencv-python", "ale-py", "stable-baselines3", "tensorboard", "torch"]
-extras = ["optuna", "numba", "ffmpeg-python", "pyvirtualdisplay", "scikit-fda", "scikit-learn"]
+extras = ["optuna", "ffmpeg-python", "scikit-fda"]
 doc = ["sphinx", "sphinx-gallery", "sphinx-math-dollar", "numpydoc", "myst-parser", "sphinxcontrib-video", "matplotlib"]
 
 [tool.poetry.group.dev]

rlberry/check_packages.py

@@ -11,9 +11,3 @@
     import torch.utils.tensorboard
 except ModuleNotFoundError:  # pragma: no cover
     TENSORBOARD_INSTALLED = False  # pragma: no cover
-
-NUMBA_INSTALLED = True
-try:
-    import numba
-except ModuleNotFoundError:  # pragma: no cover
-    NUMBA_INSTALLED = False  # pragma: no cover

rlberry/rendering/tests/test_rendering_interface.py

@@ -2,7 +2,6 @@
 import pytest
 import sys
 
-from pyvirtualdisplay import Display
 from rlberry_research.envs.classic_control import MountainCar
 from rlberry_research.envs.classic_control import Acrobot
 from rlberry_research.envs.classic_control import Pendulum
@@ -20,11 +19,6 @@
 
 import tempfile
 
-try:
-    display = Display(visible=0, size=(1400, 900))
-    display.start()
-except Exception:
-    pass
 
 classes = [
     Acrobot,

rlberry/utils/logging.py

@@ -110,5 +110,3 @@ def configure_logging(
     gym.logger.set_level(
         logging.getLevelName(level) + 10
     )  # If info -> go to warning gym level. If debug, go to info.
-    numba_logger = logging.getLogger("numba")
-    numba_logger.setLevel(logging.WARNING)