add many mock import

KindXiaoming · Apr 29, 2024 · 87e22e6 · 87e22e6
1 parent 718742a
commit 87e22e6
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/docs/conf.py b/docs/conf.py
@@ -40,7 +40,14 @@ def skip(app, what, name, obj, would_skip, options):
 def setup(app):
     app.connect("autodoc-skip-member", skip)
 
-autodoc_mock_imports = ["numpy"]
+autodoc_mock_imports = ["numpy",
+                        "torch",
+                        "torch.nn",
+                        "matplotlib.pyplot",
+                        "tqdm",
+                        "sympy",
+                        "sklearn.linear_model",
+                        "torch.optim"]
 
 
 source_suffix = [".rst", ".md"]

diff --git a/kan/KANLayer.py b/kan/KANLayer.py
@@ -2,7 +2,6 @@
 import torch.nn as nn
 import numpy as np
 from .spline import *
-import inspect
 
 
 class KANLayer(nn.Module):
@@ -367,4 +366,4 @@ def unlock(self, ids):
             self.lock_id[ids[i][1]*self.in_dim+ids[i][0]] = 0
         self.lock_counter -= 1
 
-
+