Update MLBaseModelDriver.py

MLBaseModelDriver.py

@@ -126,13 +126,16 @@ class MLBaseModelDriver:
 
         return float(predicted_sale_price), predicted_sale_date.strftime("%Y-%m-%d")
 
-    def _sale_date_predictor(self, days_on_market: int, predicted_days_on_market: int) -> datetime.date:
+    def _sale_date_predictor(self, days_on_market: Optional[int], predicted_days_on_market: int) -> datetime.date:
         """
         convert predicted days on market to a sale date
         :param days_on_market: number of days this home has been on the market
         :param predicted_days_on_market: the predicted number of days for this home on the market
         :return: the predicted sale date
         """
+        if days_on_market is None:
+            days_on_market = 0
+            
         if days_on_market < predicted_days_on_market:
             days_until_sale = predicted_days_on_market - days_on_market
             sale_date = datetime.date.today() + datetime.timedelta(days=days_until_sale)