Expose get_torch_default_device through thinc.api (#698)

explosion · Jun 15, 2022 · 0711e60 · 0711e60
1 parent 862a489
commit 0711e60
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/thinc/api.py b/thinc/api.py
@@ -16,6 +16,7 @@
 from .util import DataValidationError, data_validation
 from .util import to_categorical, get_width, get_array_module, to_numpy
 from .util import torch2xp, xp2torch, tensorflow2xp, xp2tensorflow, mxnet2xp, xp2mxnet
+from .util import get_torch_default_device
 from .compat import has_cupy
 from .backends import get_ops, set_current_ops, get_current_ops, use_ops
 from .backends import Ops, CupyOps, MPSOps, NumpyOps, set_gpu_allocator