From 02ad057073e13ef2e9b34bcd6c65529162d04e7a Mon Sep 17 00:00:00 2001 From: Arnaud Gelas Date: Thu, 7 Jan 2021 10:32:02 +0100 Subject: [PATCH] Fix pre-commit isort hook failure for benchmarks subdir. --- benchmarks/generate_comparison.py | 2 +- benchmarks/test_basic_parity.py | 2 +- benchmarks/test_sharded_parity.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/benchmarks/generate_comparison.py b/benchmarks/generate_comparison.py index 6b5a0680a6b36..d1ce6012d63ef 100644 --- a/benchmarks/generate_comparison.py +++ b/benchmarks/generate_comparison.py @@ -14,8 +14,8 @@ import os import matplotlib.pylab as plt -import pandas as pd +import pandas as pd from benchmarks.test_basic_parity import measure_loops from tests.base.models import ParityModuleMNIST, ParityModuleRNN diff --git a/benchmarks/test_basic_parity.py b/benchmarks/test_basic_parity.py index ce3d831f099f5..679d1539fa622 100644 --- a/benchmarks/test_basic_parity.py +++ b/benchmarks/test_basic_parity.py @@ -16,9 +16,9 @@ import numpy as np import pytest -import torch from tqdm import tqdm +import torch from pytorch_lightning import LightningModule, seed_everything, Trainer from tests.base.models import ParityModuleMNIST, ParityModuleRNN diff --git a/benchmarks/test_sharded_parity.py b/benchmarks/test_sharded_parity.py index 05fde8e11523a..1575ad4650a06 100644 --- a/benchmarks/test_sharded_parity.py +++ b/benchmarks/test_sharded_parity.py @@ -18,8 +18,8 @@ from typing import Type, Union import pytest -import torch +import torch from pytorch_lightning import seed_everything, Trainer from pytorch_lightning.plugins.ddp_plugin import DDPPlugin from pytorch_lightning.plugins.sharded_plugin import DDPShardedPlugin