mgbam commited on
Commit
139e188
·
verified ·
1 Parent(s): 414568a

Update tools/forecaster.py

Browse files
Files changed (1) hide show
  1. tools/forecaster.py +1 -4
tools/forecaster.py CHANGED
@@ -1,11 +1,8 @@
1
-
2
  import pandas as pd
3
  import matplotlib.pyplot as plt
4
  from statsmodels.tsa.arima.model import ARIMA
5
- from google.adk.tools import Tool
6
 
7
- @Tool(name="forecast_tool", description="Forecast future sales using ARIMA")
8
- def forecast(file_path: str) -> str:
9
  df = pd.read_csv(file_path)
10
  df['Month'] = pd.to_datetime(df['Month'])
11
  df.set_index('Month', inplace=True)
 
 
1
  import pandas as pd
2
  import matplotlib.pyplot as plt
3
  from statsmodels.tsa.arima.model import ARIMA
 
4
 
5
+ def forecast_tool(file_path: str) -> str:
 
6
  df = pd.read_csv(file_path)
7
  df['Month'] = pd.to_datetime(df['Month'])
8
  df.set_index('Month', inplace=True)