Update main.py

main.py

@@ -41,8 +41,10 @@ def read_root():
 
 # XGboost
 
+
+ 
 def data_download(ticker: str):
-    ticker= ticker
+    ticker = ticker
     # Define the list of tickers
     index_list = [
         ticker, '^VIX', '^VVIX', '^VIX9D', '^VIX3M', '^VIX6M', '^FVX', '^TNX', '^TYX'
@@ -65,7 +67,8 @@ def data_download(ticker: str):
 
 
     
-def data_manipolation(df):
+def data_manipolation(df,ticker: str):
+    ticker = ticker
     # MA calculation for all columns 
     New_Names=[ticker,'VIX','VIX_Index','VIX9D','VIX3M','VIX6M','T5Y','T10Y','T30Y']
     for col in New_Names:
@@ -97,7 +100,8 @@ def data_manipolation(df):
 
     return df
        
-def data_split_train_test(df):
+def data_split_train_test(df,ticker: str):
+    ticker = ticker
     X = df.loc[:,df.columns !=  ticker + '_c1']
     y = df[ ticker + '_c1']
     recent_data_size = int(0.3 * len(X))  # Adjust the percentage as needed