Update pages/Tyre_Strategy.py

pages/Tyre_Strategy.py

@@ -1,5 +1,5 @@
 import streamlit as st
-from repo_directory import Tyre_Strategy as ts
+from repo_directory import Tyre_Strategy
 from repo_directory import button
 
 
@@ -11,8 +11,8 @@ YEAR_SELECTED = st.selectbox(
 RACE_SELECTED =  st.selectbox(
     'Select Race',
     (1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22))
-SESSION = "RACE"
 
-driver_stints, drivers_list = ts.get_data(YEAR_SELECTED, RACE_SELECTED)
 
-ts.plot(driver_stints,drivers_list )
+driver_stints, drivers_list = Tyre_Strategy.get_data(YEAR_SELECTED, RACE_SELECTED)
+
+Tyre_Strategy.plot(driver_stints,drivers_list )