tried to fix some things, script is still broken

1 files changed, 1 insertions, 1 deletions

diff --git a/solver2.py b/solver2.py
index ae2403a..906673c 100644
--- a/solver2.py
+++ b/solver2.py
@@ -45,7 +45,7 @@ class Learner(object):
 		if args.end != None:
 			confirmed_sums = np.sum([reg.loc[args.start:args.end].values for reg in country_df.iloc], axis = 0)
 		else:
-			confirmed_sums = np.sum([reg.loc[[args.start:]].values for reg in country_df.iloc], axis = 0)
+			confirmed_sums = np.sum([reg.loc[args.start:].values for reg in country_df.iloc], axis = 0)
 		
 		if args.end != None:
 			new_data = pd.DataFrame(confirmed_sums, country_df.iloc[0].loc[args.start:args.end].index.tolist())