Jitendra12421 commited on
Commit
f76aebb
·
verified ·
1 Parent(s): a3b7b62

Upload forecaster_cli.py

Browse files
Files changed (1) hide show
  1. forecaster_cli.py +9 -1
forecaster_cli.py CHANGED
@@ -159,13 +159,21 @@ def get_prediction(ticker):
159
  decision = "BUY" if prob > 0.65 else "PASS"
160
  reasoning = generate_reasoning(features, tier, prob)
161
 
 
 
 
 
 
 
 
 
162
  return {
163
  "Ticker": ticker,
164
  "Tier": tier,
165
  "Decision": decision,
166
  "Confidence": prob,
167
  "Reasoning": reasoning,
168
- "Features": features
169
  }
170
 
171
  def run_inference(ticker):
 
159
  decision = "BUY" if prob > 0.65 else "PASS"
160
  reasoning = generate_reasoning(features, tier, prob)
161
 
162
+ import math
163
+ clean_features = {}
164
+ for k, v in features.items():
165
+ if isinstance(v, float) and math.isnan(v):
166
+ clean_features[k] = None
167
+ else:
168
+ clean_features[k] = v
169
+
170
  return {
171
  "Ticker": ticker,
172
  "Tier": tier,
173
  "Decision": decision,
174
  "Confidence": prob,
175
  "Reasoning": reasoning,
176
+ "Features": clean_features
177
  }
178
 
179
  def run_inference(ticker):