diff --git a/pyproject.toml b/pyproject.toml
index 5155c92022b4..649ea8246cea 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -161,8 +161,7 @@ omit = [
 exclude_lines = [
     "pragma: no cover",
     "pass",
-    "raise",
-    "except",
+    "raise NotImplementedError",
     "register_parameter",
     "warn",
     "torch.cuda.is_available",