From aa1f6dba7352f71d2c46575461fef23423b7bd4a Mon Sep 17 00:00:00 2001 From: Fabrice Normandin Date: Tue, 18 Jun 2024 12:01:00 -0400 Subject: [PATCH] Minor touchup Signed-off-by: Fabrice Normandin --- project/datamodules/datamodules_test.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/project/datamodules/datamodules_test.py b/project/datamodules/datamodules_test.py index 5667990e..8737df7d 100644 --- a/project/datamodules/datamodules_test.py +++ b/project/datamodules/datamodules_test.py @@ -3,7 +3,10 @@ import matplotlib.pyplot as plt import pytest -from tensor_regression.fixture import TensorRegressionFixture, get_test_source_and_temp_file_paths +from tensor_regression.fixture import ( + TensorRegressionFixture, + get_test_source_and_temp_file_paths, +) from torch import Tensor from project.utils.testutils import run_for_all_datamodules @@ -68,7 +71,10 @@ def test_first_batch( fig.suptitle(f"First batch of datamodule {type(datamodule).__name__}") figure_path, _ = get_test_source_and_temp_file_paths( - extension=".png", request=request, original_datadir=original_datadir, datadir=datadir + extension=".png", + request=request, + original_datadir=original_datadir, + datadir=datadir, ) figure_path.parent.mkdir(exist_ok=True, parents=True) fig.savefig(figure_path)