From 690bd155086bf737bf24c3ead369cd7ad68efd9d Mon Sep 17 00:00:00 2001
From: Andre Anjos <andre.dos.anjos@gmail.com>
Date: Tue, 19 Sep 2023 14:00:37 +0200
Subject: [PATCH] [data.datamodule] Fix warning and object setup
---
src/ptbench/data/datamodule.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/src/ptbench/data/datamodule.py b/src/ptbench/data/datamodule.py
index 82c2d19a..8b158c28 100644
--- a/src/ptbench/data/datamodule.py
+++ b/src/ptbench/data/datamodule.py
@@ -528,11 +528,11 @@ class ConcatDataModule(lightning.LightningDataModule):
self.parallel = parallel # immutable, otherwise would need to call
self.pin_memory = (
- torch.cuda.is_available() or torch.backends.mps.is_available()
+ torch.cuda.is_available() or torch.backends.mps.is_available() # type: ignore
) # should only be true if GPU available and using it
# datasets that have been setup() for the current stage
- self._datasets: CachingDataModule.DatasetDictionary = {}
+ self._datasets: ConcatDataModule.DatasetDictionary = {}
@property
def parallel(self) -> int:
--
GitLab