Sivainti commited on
Commit
2f0cfaa
·
verified ·
1 Parent(s): 4e2b778

Update utils/_init_.py

Browse files
Files changed (1) hide show
  1. utils/_init_.py +0 -15
utils/_init_.py CHANGED
@@ -1,15 +0,0 @@
1
- import pandas as pd
2
- from datetime import timedelta
3
-
4
- def predict_timeline(df):
5
- # Dummy logic for testing
6
- df['Start'] = pd.Timestamp.now()
7
- df['Finish'] = df['Start'] + pd.to_timedelta(7, unit='D')
8
- df['Task'] = df.iloc[:, 0] # Assume first column is task name
9
- df['Risk'] = ['Low'] * len(df)
10
-
11
- summary = "Timeline generated for {} tasks.".format(len(df))
12
- risks = [{"text": task, "start": 0, "end": len(task), "entity": risk}
13
- for task, risk in zip(df['Task'], df['Risk'])]
14
-
15
- return df[['Task', 'Start', 'Finish', 'Risk']], risks, summary