From 29e3cf5579b95117a2cebd4275586058c0512a34 Mon Sep 17 00:00:00 2001
From: "Bryn N. Ubald" <55503826+bnubald@users.noreply.github.com>
Date: Wed, 29 May 2024 11:39:30 +0100
Subject: [PATCH] Fixes #265: Include missing torch load bool

---
 icenet/data/dataset.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/icenet/data/dataset.py b/icenet/data/dataset.py
index 8763697..d47cc10 100644
--- 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: