diff --git a/docs/source/conf.py b/docs/source/conf.py
index 8a4a833f..33a34353 100644
--- a/docs/source/conf.py
+++ b/docs/source/conf.py
@@ -68,4 +68,4 @@
 add_module_names = False
 
 autodoc_mock_imports = ['sklearn',  'pandas', 'numpy', 'datamart_profiler', 'PipelineProfiler', 'IPython', 'torch',
-                        'nltk', 'xgboost']
+                        'nltk', 'xgboost', 'ray', 'gymnasium']