Fixes icenet-ai#265: Include missing torch load bool

bnubald · May 29, 2024 · 29e3cf5 · 29e3cf5
1 parent 662882f
commit 29e3cf5
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/icenet/data/dataset.py b/icenet/data/dataset.py
@@ -18,6 +18,7 @@
 pytorch_available = False
 try:
     from torch.utils.data import Dataset
+    pytorch_available = True
 except ModuleNotFoundError:
     print("PyTorch not found - not required if not using PyTorch")
 except ImportError: