Update MLBaseModelDriver.py

MLBaseModelDriver.py

@@ -1,14 +1,12 @@
 import torch
 import sys
 import pandas as pd
-pd.set_option('future.no_silent_downcasting', True)
 from typing import TypedDict, Optional, Tuple
 import datetime
 import math
 import importlib.util
 from huggingface_hub import hf_hub_download
 import pickle
-import time
 
 
 """
@@ -151,15 +149,11 @@ class MLBaseModelDriver:
         default_beds = 3
         default_sqft = 1500.0
         default_property_type = '6'
-    
-        # Fill and type inference
         df['beds'] = df['beds'].fillna(default_beds).infer_objects(copy=False).astype(int)
         df['sqft'] = pd.to_numeric(df['sqft'], errors='coerce').fillna(default_sqft)
-        df['property_type'] = df['property_type'].fillna(default_property_type).infer_objects(copy=False)
+        df['property_type'] = df['property_type'].fillna(default_property_type)
         df['property_type'] = df['property_type'].astype(int)
-    
-        # Normalize
         df[['sqft', 'price']] = self.scaler.transform(df[['sqft', 'price']])
         X = df[['beds', 'sqft', 'property_type', 'price']]
         input_tensor = torch.tensor(X.values, dtype=torch.float32)
-        return input_tensor
+        return input_tensor