YaakovY commited on
Commit
6479df0
1 Parent(s): 2262d7f
Files changed (1) hide show
  1. main.py +2 -1
main.py CHANGED
@@ -2,6 +2,7 @@ from fastapi import FastAPI
2
  import pandas as pd
3
  import numpy as np
4
  import matplotlib.pyplot as plt
 
5
 
6
  # import seaborn as sns
7
  import yfinance as yf
@@ -242,4 +243,4 @@ def prcess_ticker(ticker: str):
242
  df = create_features(df)
243
  result = run_xgboost(df)
244
 
245
- return result
 
2
  import pandas as pd
3
  import numpy as np
4
  import matplotlib.pyplot as plt
5
+ import json
6
 
7
  # import seaborn as sns
8
  import yfinance as yf
 
243
  df = create_features(df)
244
  result = run_xgboost(df)
245
 
246
+ return json.dumps(result, cls=NumpyEncoder)