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: