From b5a49ca89d1305707bd5c1485aa7890deb486452 Mon Sep 17 00:00:00 2001
From: Philip Abbet <philip.abbet@idiap.ch>
Date: Thu, 14 Dec 2023 16:38:19 +0100
Subject: [PATCH] [data.datamodule] Fix incomplete teardown of DataModule
---
src/ptbench/data/datamodule.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/ptbench/data/datamodule.py b/src/ptbench/data/datamodule.py
index 8b158c28..e1ba75a3 100644
--- a/src/ptbench/data/datamodule.py
+++ b/src/ptbench/data/datamodule.py
@@ -815,7 +815,7 @@ class ConcatDataModule(lightning.LightningDataModule):
* ``test``: uses only the test dataset
* ``predict``: uses only the test dataset
"""
- pass
+ super().teardown(stage)
def train_dataloader(self) -> DataLoader:
"""Returns the train data loader."""
--
GitLab