diff --git a/src/awkward/operations/ak_to_jaggedtensor.py b/src/awkward/operations/ak_to_jaggedtensor.py index bc222afc3b..43c7cb1501 100644 --- a/src/awkward/operations/ak_to_jaggedtensor.py +++ b/src/awkward/operations/ak_to_jaggedtensor.py @@ -44,6 +44,7 @@ def to_jaggedtensor( def _impl(array, padded, padding_value, max_lengths, keep_regular): try: + import fbgemm_gpu import torch except ImportError as err: raise ImportError(