macos: reduce precision to float32

Handle: Cannot convert a MPS Tensor to float64 dtype as the MPS framework doesn't support float64. Please use float32 instead

Author: 2maz

src/damast/data_handling/accessors.py

@@ -4,6 +4,7 @@
 
 import logging
 import random
+import sys
 import time
 from typing import Any, List, Optional, Union
 
@@ -22,6 +23,16 @@
 logger = logging.getLogger("damast")
 
 
+if sys.platform == "darwin":
+    # Handle "Cannot convert a MPS Tensor to float64 dtype as the MPS framework doesn't support float64. Please use float32 instead"
+    def _mps_precision(data):
+        if data.dtype == np.float64:
+            return data.astype(np.float32)
+        return data
+else:
+    def _mps_precision(data):
+        return data
+
 # https://www.tensorflow.org/tutorials/structured_data/time_series
 class GroupSequenceAccessor:
     """
@@ -289,12 +300,12 @@ def _generator(features: List[str], target: Optional[List[str]],
                         # target it the last step in the timeline, so the last
                         target_chunk.append(target_window.to_numpy())
 
-                X = np.array(chunk)
+                X = _mps_precision(np.array(chunk))
                 if use_target:
                     if np.lib.NumpyVersion(np.__version__) >= '2.0.0':
-                        y = np.array(target_chunk)
+                        y = _mps_precision(np.array(target_chunk))
                     else:
-                        y = np.array(target_chunk, copy=False)
+                        y = _mps_precision(np.array(target_chunk, copy=False))
                     yield (X, y)
                 else:
                     yield (X,)