From 99ae8e1dbedaebf77415302031469a54d69bf9c0 Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Sun, 29 Sep 2024 18:41:27 +0000
Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
---
 tests/test_data.py     | 1 +
 tests/test_finetune.py | 1 +
 tests/test_model.py    | 1 +
 tests/test_pooling.py  | 1 +
 4 files changed, 4 insertions(+)

diff --git a/tests/test_data.py b/tests/test_data.py
index bb1a37a..3a5658d 100644
--- a/tests/test_data.py
+++ b/tests/test_data.py
@@ -4,6 +4,7 @@
 import torch
 
 from smtb.data import DownstreamDataModule, DownstreamDataset
+
 from .fixtures import mock_data_dir
 
 
diff --git a/tests/test_finetune.py b/tests/test_finetune.py
index b4d0517..5d6966e 100644
--- a/tests/test_finetune.py
+++ b/tests/test_finetune.py
@@ -1,6 +1,7 @@
 import os
 
 from smtb.train import train
+
 from .fixtures import mock_data_dir, sample_config
 
 
diff --git a/tests/test_model.py b/tests/test_model.py
index d6b09fe..35280ae 100644
--- a/tests/test_model.py
+++ b/tests/test_model.py
@@ -4,6 +4,7 @@
 import torch
 
 from smtb.model import RegressionModel
+
 from .fixtures import sample_batch_x, sample_config
 
 
diff --git a/tests/test_pooling.py b/tests/test_pooling.py
index 8454506..c622f4f 100644
--- a/tests/test_pooling.py
+++ b/tests/test_pooling.py
@@ -4,6 +4,7 @@
 import torch
 
 from smtb.model import poolings
+
 from .fixtures import sample_batch_x, sample_config