Skip to content

Commit c4b1dd9

Browse files
committed
Fixed what I screwed up with timeparameters.py
1 parent 1703f0d commit c4b1dd9

File tree

1 file changed

+6
-7
lines changed

1 file changed

+6
-7
lines changed

paramgenerator/timeparameters.py

Lines changed: 6 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -143,18 +143,17 @@ def computeTimeMedians(factors, lastmonthcount = LAST_MONTHS):
143143

144144
return (medianFirstMonth, medianLastMonth, median)
145145

146-
def readTimeParams(persons, factorFiles, friendFiles):
146+
def readTimeParams(persons, personFactorFiles, activityFactorFiles, friendFiles):
147147

148148
postCounts = {}
149149
groupCounts = {}
150150
offset = 8
151151
monthcount = 12*3 + 1
152152

153-
for inputFactorFile in factorFiles:
153+
for inputFactorFile in personFactorFiles:
154154
with open(inputFactorFile, 'r') as f:
155-
personCount = int(f.readline())
156-
for i in range(personCount):
157-
line = f.readline().split(",")
155+
for line in f.readlines():
156+
line = line.split(",")
158157
person = int(line[0])
159158
localPostCounts = map(int,line[offset:offset+monthcount])
160159
localGroupCounts = map(int, line[offset+monthcount:])
@@ -209,15 +208,15 @@ def readTimeParams(persons, factorFiles, friendFiles):
209208

210209

211210

212-
def findTimeParams(input, factorFiles, friendFiles, startYear):
211+
def findTimeParams(input, personFactorFiles, activityFactorFiles, friendFiles, startYear):
213212
START_YEAR = startYear
214213
fPostCount = {}
215214
ffPostCount = {}
216215
persons = []
217216
for queryId in input:
218217
persons += input[queryId][0]
219218

220-
(fPostCount, ffPostCount, ffGroupCount) = readTimeParams(set(persons),factorFiles, friendFiles)
219+
(fPostCount, ffPostCount, ffGroupCount) = readTimeParams(set(persons),personFactorFiles, activityFactorFiles, friendFiles)
221220

222221
mapParam = {
223222
"f" : fPostCount,

0 commit comments

Comments
 (0)