save date in json, fixes (#651)

mljar · Sep 26, 2023 · 2a22f03 · 2a22f03
1 parent 5775bac
commit 2a22f03
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/supervised/utils/jsonencoder.py b/supervised/utils/jsonencoder.py
@@ -1,6 +1,6 @@
 import json
 import numpy as np
-
+from datetime import date
 
 class MLJSONEncoder(json.JSONEncoder):
     def default(self, o):
@@ -25,5 +25,7 @@ def default(self, o):
             return float(o)
         elif isinstance(o, np.ndarray):
             return o.tolist()
+        elif isinstance(obj, date):
+            return obj.strftime("%Y-%m-%d")
 
-        return super().default(o)
+        return super(MLJSONEncoder, self).default(o)