From a263aa3ecd664cd6eb97b09bc141191534752e9d Mon Sep 17 00:00:00 2001
From: Nicholas Cilfone <nicholas.cilfone@fmr.com>
Date: Fri, 8 Jul 2022 10:20:22 -0400
Subject: [PATCH] whoops

---
 stoke/data.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/stoke/data.py b/stoke/data.py
index 946f58a..27878b3 100644
--- a/stoke/data.py
+++ b/stoke/data.py
@@ -105,7 +105,7 @@ def _place_data_on_gpu(
 
         """
         # Call to the public function in utils
-        place_data_on_gpu(data, self._fp16)
+        return place_data_on_gpu(data, self._fp16)
 
 
 class BucketedDistributedSampler(Sampler[T_co]):