Skip to content

Commit

Permalink
Update TransactionalDatabase.py
Browse files Browse the repository at this point in the history
  • Loading branch information
Tarun-Sreepada committed May 7, 2024
1 parent 072c662 commit e4646c9
Showing 1 changed file with 6 additions and 2 deletions.
8 changes: 6 additions & 2 deletions PAMI/extras/syntheticDataGenerator/TransactionalDatabase.py
Original file line number Diff line number Diff line change
Expand Up @@ -208,8 +208,12 @@ def getTransactions(self) -> pd.DataFrame:
Returns:
db: pd.dataFrame - transactional database
"""
df = pd.DataFrame(self.db)
return df
column = "Transactions"
db = pd.DataFrame(columns=[column])
self.db = [tuple(x) for x in self.db]
for i in range(len(self.db)):
db.at[i,column] = self.db[i]
return db


if __name__ == "__main__":
Expand Down

0 comments on commit e4646c9

Please sign in to comment.