Fix Read the Docs build configuration

.readthedocs.yaml

@@ -18,13 +18,9 @@ build:
   os: ubuntu-22.04
   tools:
     python: "3.11"
-  jobs:
-    post_install:
-      # Install package dependencies
-      - pip install -e .
 
 python:
   install:
-    - requirements: docs/requirements.txt
     - method: pip
       path: .
+    - requirements: docs/requirements.txt

docs/source/conf.py

@@ -60,6 +60,17 @@
 autodoc_typehints = "description"
 autodoc_class_signature = "separated"
 
+# Mock imports for heavy dependencies that aren't needed for building docs
+autodoc_mock_imports = [
+    "torch",
+    "torch_geometric",
+    "pytorch_lightning",
+    "torchmetrics",
+    "tensorflow",
+    "keras",
+    "spektral",
+]
+
 # Add any paths that contain templates here, relative to this directory.
 templates_path = ["_templates"]
 

unravel/classifiers/__init__.py

@@ -14,6 +14,17 @@ def __init__(self, *args, **kwargs):
     __all__ = ["CrystalGraphClassifier"]
 
 
-from .crystal_graph_pyg import PyGLightningCrystalGraphClassifier
+try:
+    from .crystal_graph_pyg import PyGLightningCrystalGraphClassifier
+
+    __all__.append("PyGLightningCrystalGraphClassifier")
+except ImportError:
+
+    class PyGLightningCrystalGraphClassifier:
+        def __init__(self, *args, **kwargs):
+            raise ImportError(
+                "PyGLightningCrystalGraphClassifier requires PyTorch and PyTorch Geometric. "
+                "Install with: pip install torch torch-geometric pytorch-lightning torchmetrics"
+            )
 
-__all__.append("PyGLightningCrystalGraphClassifier")
+    __all__.append("PyGLightningCrystalGraphClassifier")