From cf87696fa60d5ad4b36509f874318e7f80df4b24 Mon Sep 17 00:00:00 2001 From: jalving Date: Sun, 8 Dec 2024 08:38:22 -0800 Subject: [PATCH] fix ruff --- .../neuralnet/graph_neural_network_formulation.ipynb | 5 +++-- pyproject.toml | 1 - 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/docs/notebooks/neuralnet/graph_neural_network_formulation.ipynb b/docs/notebooks/neuralnet/graph_neural_network_formulation.ipynb index 242a02ac..e30c94f7 100644 --- a/docs/notebooks/neuralnet/graph_neural_network_formulation.ipynb +++ b/docs/notebooks/neuralnet/graph_neural_network_formulation.ipynb @@ -338,8 +338,9 @@ "X = np.array(X).reshape(3, 2)\n", "edges = np.transpose(np.array(edges)).reshape(2, -1)\n", "nn.eval()\n", - "# TODO: update for latest torch.geometric\n", - "# print(nn1(torch.tensor(X).float(), torch.tensor(edges)).detach().numpy())" + "# TODO @zshiqiang: update for latest torch.geometric # noqa: FIX002\n", + "# https://github.com/cog-imperial/OMLT/issues/166\n", + "# print(nn1(torch.tensor(X).float(), torch.tensor(edges)).detach().numpy()) : noqa: ERA001" ] }, { diff --git a/pyproject.toml b/pyproject.toml index 85ada492..b10b1a84 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -74,7 +74,6 @@ extend-exclude = ["src/omlt/_version.py"] [tool.ruff.lint] select = ["ALL"] ignore = [ - "ANN101", "ANN401", "COM812", "ISC001",