From d74ad225301d7051f4f2b67e887ca279d88c4883 Mon Sep 17 00:00:00 2001
From: David Bau <david.bau@gmail.com>
Date: Tue, 23 Aug 2022 06:08:59 -0400
Subject: [PATCH] Fix move_to on lists.

---
 baukit/tokendataset.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/baukit/tokendataset.py b/baukit/tokendataset.py
index 3715c8e..c704671 100644
--- a/baukit/tokendataset.py
+++ b/baukit/tokendataset.py
@@ -47,7 +47,7 @@ def move_to(device, *containers):
             if isinstance(container, dict):
                 g = list(container.items())
             else:
-                g = enumerate(list)
+                g = enumerate(container)
             for i, v in g:
                 moved = move_to(device, v)
                 if moved is not None: