diff --git a/docs/conf.py b/docs/conf.py
index b208ff37..559f3e15 100644
--- 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
index bd1a35eb..1cd5dbcd 100644
--- 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
 
-    
\ No newline at end of file
+