Mercurial Hosting > traffic-intelligence
diff trafficintelligence/storage.py @ 1049:c9c03c97ed9f
bug fix to store numpy integers in SQLite
author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
---|---|
date | Fri, 06 Jul 2018 15:58:45 -0400 |
parents | 75a6ad604cc5 |
children | 9d4a06f49cb8 d13f9bfbf3ff |
line wrap: on
line diff
--- a/trafficintelligence/storage.py Fri Jul 06 11:13:10 2018 -0400 +++ b/trafficintelligence/storage.py Fri Jul 06 15:58:45 2018 -0400 @@ -7,7 +7,7 @@ from copy import copy import sqlite3, logging -from numpy import log, min as npmin, max as npmax, round as npround, array, sum as npsum, loadtxt, floor as npfloor, ceil as npceil, linalg +from numpy import log, min as npmin, max as npmax, round as npround, array, sum as npsum, loadtxt, floor as npfloor, ceil as npceil, linalg, int32, int64 from pandas import read_csv, merge from trafficintelligence import utils, moving, events, indicators @@ -22,6 +22,9 @@ 'object': 'objects', 'objectfeatures': 'positions'} +sqlite3.register_adapter(int64, lambda val: int(val)) +sqlite3.register_adapter(int32, lambda val: int(val)) + ######################### # Sqlite #########################