Fix _XLA_AVAILABLE import

Author: freud14

poutyne/framework/metrics/predefined/bincount.py

@@ -39,7 +39,12 @@
 
 import torch
 from torch import Tensor
-from torchmetrics.utilities.imports import _TORCH_GREATER_EQUAL_1_12, _XLA_AVAILABLE
+from torchmetrics.utilities.imports import _TORCH_GREATER_EQUAL_1_12
+
+try:
+    from torchmetrics.utilities.imports import _XLA_AVAILABLE
+except ImportError:
+    _XLA_AVAILABLE = False
 
 
 def _bincount(x: Tensor, minlength: Optional[int] = None) -> Tensor: