Skip to content

Commit

Permalink
fix: noqa
Browse files Browse the repository at this point in the history
  • Loading branch information
aMahanna committed May 29, 2024
1 parent 600c02b commit 3f0e2f5
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions python/phenolrs/pyg_loader.py
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down Expand Up @@ -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:
Expand Down

0 comments on commit 3f0e2f5

Please sign in to comment.