Fix serialization errors.

Author: isaacrobinson2000

diplomat/__init__.py

@@ -2,7 +2,7 @@
 A tool providing multi-animal tracking capabilities on top of other Deep learning based tracking software.
 """
 
-__version__ = "0.3.1"
+__version__ = "0.3.2"
 # Can be used by functions to determine if diplomat was invoked through it's CLI interface.
 CLI_RUN = False
 

diplomat/predictors/fpe/frame_passes/optimize_std.py

@@ -51,7 +51,7 @@ def run_pass(
         result = super().run_pass(fb_data, prog_bar, in_place, reset_bar)
 
         approx_std = (self._histogram.get_quantile(0.5)[2] / self.MAGIC_CONST) * self.config.std_multiplier
-        result.metadata.optimal_std = (*self._histogram.get_bin_for_value(approx_std)[:2], approx_std)
+        result.metadata.optimal_std = (*self._histogram.get_bin_for_value(approx_std)[:2], float(approx_std))
 
         if(self.config.DEBUG):
             print(f"Optimal STD: {result.metadata.optimal_std}")

diplomat/predictors/fpe/skeleton_structures.py

@@ -302,7 +302,7 @@ def add(self, value: float, weight: int = 1):
     def get_bin_for_value(self, value: float) -> Tuple[float, int, float]:
         val_bin = int((value - self._bin_offset) / self._bin_size)
         freq, avg = self._bins.get(val_bin, (0, value))
-        return (val_bin * self._bin_size + self._bin_offset, freq, avg)
+        return (float(val_bin * self._bin_size + self._bin_offset), int(freq), float(avg))
 
     def __iter__(self) -> Iterable[float]:
         return (b * self._bin_size + self._bin_offset for b in self._bins)

diplomat/predictors/sfpe/file_io.py

@@ -366,8 +366,10 @@ def _is_fully_covered_no_overlap(self):
     def _encode_meta_chunk(self, data: dict = None) -> bytes:
         if(data is None):
             data = {}
-
-        return zlib.compress(json.dumps(data, cls=FPEMetadataEncoder).encode(), self._compression_level)
+        try:
+            return zlib.compress(json.dumps(data, cls=FPEMetadataEncoder).encode(), self._compression_level)
+        except TypeError as e:
+            raise ValueError(f"Bad metadata object: {data}") from e
 
     def _decode_meta_chunk(self, data: bytes) -> dict:
         if(len(data) == 0):