diff --git a/python/phenolrs/pyg_loader.py b/python/phenolrs/pyg_loader.py
index d4f6ec5..5a7bca7 100644
--- a/python/phenolrs/pyg_loader.py
+++ b/python/phenolrs/pyg_loader.py
@@ -28,7 +28,7 @@ def load_into_pyg_data(
         batch_size: int | None = None,
     ) -> tuple["Data", dict[str, dict[str, int]], dict[str, dict[int, str]]]:
         if not TORCH_AVAILABLE:
-            m = "Missing required dependencies. Install with `pip install phenolrs[torch]`"
+            m = "Missing required dependencies. Install with `pip install phenolrs[torch]`"  # noqa: E501
             raise ImportError(m)
 
         if "vertexCollections" not in metagraph:
@@ -111,7 +111,7 @@ def load_into_pyg_heterodata(
         batch_size: int | None = None,
     ) -> tuple["HeteroData", dict[str, dict[str, int]], dict[str, dict[int, str]]]:
         if not TORCH_AVAILABLE:
-            m = "Missing required dependencies. Install with `pip install phenolrs[torch]`"
+            m = "Missing required dependencies. Install with `pip install phenolrs[torch]`"  # noqa: E501
             raise ImportError(m)
 
         if "vertexCollections" not in metagraph: