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
 #########################