Abhi10000 commited on
Commit
f70fb20
·
verified ·
1 Parent(s): fedb011

Update src/model_predictor.py

Browse files
Files changed (1) hide show
  1. src/model_predictor.py +7 -4
src/model_predictor.py CHANGED
@@ -1,6 +1,9 @@
1
- from datetime import datetime, timedelta
2
 
3
- # Fake predictor: always adds 60 days
4
  def predict_next_service(last_service_date):
5
- today = datetime.today()
6
- return (today + timedelta(days=60)).strftime("%Y-%m-%d")
 
 
 
 
 
1
+ from datetime import timedelta
2
 
 
3
  def predict_next_service(last_service_date):
4
+ """
5
+ Predicts the next AMC service date by adding 180 days (6 months)
6
+ to the last service date.
7
+ """
8
+ next_service_date = last_service_date + timedelta(days=180)
9
+ return next_service_date.strftime("%Y-%m-%d")