diff --git a/.github/workflows/lint.yml b/.github/workflows/lint.yml
index e0486ae1..973612d2 100644
--- a/.github/workflows/lint.yml
+++ b/.github/workflows/lint.yml
@@ -11,4 +11,6 @@ jobs:
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v3
-      - uses: chartboost/ruff-action@v1
\ No newline at end of file
+      - uses: chartboost/ruff-action@v1
+        with:
+          src: './topobenchmarkx'
\ No newline at end of file
diff --git a/__init__.py b/__init__.py
index 37fdbef9..e151f538 100644
--- a/__init__.py
+++ b/__init__.py
@@ -1,17 +1,13 @@
+# numpydoc ignore=GL08
 import test
 
 import configs
 import topobenchmarkx
-import topobenchmarkx.nn.backbones
-
-from .topobenchmarkx import run
 
 __all__ = [
     "topobenchmarkx",
-    "topobenchmarkx.nn.backbones",
     "configs",
     "test",
-    "run",
 ]
 
 __version__ = "0.0.1"