Merge pull request #113 from Quantmetry/fix_doc

JulienRoussel77 · web-flow · commit 094878037149 · 2023-10-31T11:59:40.000+01:00
Fix doc
diff --git a/.readthedocs.yml b/.readthedocs.yml
@@ -9,6 +9,8 @@ python:
   install:
     - method: pip
       path: .
+      extra_requirements:
+        - pytorch
 
 conda:
   environment: environment.doc.yml
diff --git a/qolmat/imputations/diffusions/__init__.py b/qolmat/imputations/diffusions/__init__.py
diff --git a/qolmat/imputations/imputers_pytorch.py b/qolmat/imputations/imputers_pytorch.py
@@ -9,7 +9,6 @@
 
 from qolmat.imputations.imputers import _Imputer, ImputerRegressor
 from qolmat.utils.exceptions import EstimatorNotDefined, PyTorchExtraNotInstalled
-from qolmat.imputations.diffusions.ddpms import TabDDPM, TsDDPM
 from qolmat.benchmark import metrics
 
 try:
