Update restful/onnx_utilities.py

restful/onnx_utilities.py

@@ -59,7 +59,8 @@ class Utilities:
             denorm_price = (value * (max_close - min_close)) + min_close
         
             last_date += pd.Timedelta(days=1)
-            predicted_prices[last_date.strftime('%Y-%m-%d')] = float(denorm_price)
+            # predicted_prices[last_date.strftime('%Y-%m-%d')] = float(denorm_price)
+            predicted_prices[last_date] = self.truncate_2_decimal(denorm_price)
         
             lst_seq = np.roll(lst_seq, shift=-1, axis=1)
             lst_seq[:, -1, -1] = value
@@ -69,7 +70,7 @@ class Utilities:
         #     for date, price in predicted_prices.items()
         # ]
         predictions = [
-            {'date': date, 'price': self.truncate_2_decimal(price)}
+            {'date': date.strftime('%Y-%m-%d'), 'price': price}
             for date, price in predicted_prices.items()
         ]