From 344947ece937895a843d7cec585606c7f797b214 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Wed, 17 Jan 2024 18:39:27 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- torch_geometric/nn/dense/linear.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/torch_geometric/nn/dense/linear.py b/torch_geometric/nn/dense/linear.py index 6490ca1e83d1..55afe0bb382d 100644 --- a/torch_geometric/nn/dense/linear.py +++ b/torch_geometric/nn/dense/linear.py @@ -1,5 +1,6 @@ import copy import math +import sys import time from typing import Any, Dict, Optional, Union @@ -15,7 +16,7 @@ from torch_geometric.typing import pyg_lib from torch_geometric.utils import index_sort from torch_geometric.utils.sparse import index2ptr -import sys + if "pytest" in sys.modules: MEASURE_ITER = 1 else: