From e84f93c0760339e75cbe18d10908b29589a50a6c Mon Sep 17 00:00:00 2001
From: Nicholas Lubbers <56895592+lubbersnick@users.noreply.github.com>
Date: Fri, 5 Apr 2024 16:39:51 -0600
Subject: [PATCH] fix typo in example code

---
 examples/feature_extraction.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/examples/feature_extraction.py b/examples/feature_extraction.py
index a3bfb1a3..acd96fef 100644
--- a/examples/feature_extraction.py
+++ b/examples/feature_extraction.py
@@ -8,7 +8,7 @@
 import torch
 
 if torch.cuda.is_available():
-    device = torch.cuda.get_default_device
+    device = torch.device(0)
 else:
     device = torch.device("cpu")
 import hippynn