Mercurial Hosting > traffic-intelligence
changeset 1141:3e0f43edb4d6
Issue loading from different location
author | Etienne Beauchamp <etienne-2.beauchamp@polymtl.ca> |
---|---|
date | Thu, 16 Apr 2020 11:24:35 -0400 |
parents | 78dddfe7aa0f |
children | b3ee75b4978a |
files | trafficintelligence/storage.py |
diffstat | 1 files changed, 5 insertions(+), 0 deletions(-) [+] |
line wrap: on
line diff
--- a/trafficintelligence/storage.py Wed Apr 15 00:18:35 2020 -0400 +++ b/trafficintelligence/storage.py Thu Apr 16 11:24:35 2020 -0400 @@ -6,6 +6,7 @@ import shutil from copy import copy import sqlite3, logging +from os import chdir 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 @@ -661,6 +662,9 @@ 'Loads prototype ids and matchings (if stored)' prototypes = [] if Path(filename).is_file(): + filename = str(Path(filename).resolve()) + pwd = Path.cwd() + chdir(Path(filename).parent) with sqlite3.connect(filename) as connection: cursor = connection.cursor() objects = [] @@ -685,6 +689,7 @@ printDBError(error) if len(set([p.getTrajectoryType() for p in prototypes])) > 1: print('Different types of prototypes in database ({}).'.format(set([p.getTrajectoryType() for p in prototypes]))) + chdir(pwd) return prototypes def savePOIsToSqlite(filename, gmm, gmmType, gmmId):