Merge remote-tracking branch 'origin/maxymnaumchyk/add-to-from-jagged…

…tensor-functions' into maxymnaumchyk/add-to-from-jaggedtensor-functions # Conflicts: # src/awkward/operations/ak_to_jaggedtensor.py
scikit-hep · Sep 17, 2024 · c0da418 · c0da418
2 parents 4674108 + 9feb758
commit c0da418
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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(