diff scripts/learn-motion-patterns.py @ 998:933670761a57

updated code to python 3 (tests pass and scripts run, but non-executed parts of code are probably still not correct)
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Sun, 27 May 2018 23:22:48 -0400
parents 23f98ebb113f
children cc5cb04b04b0
line wrap: on
line diff
--- a/scripts/learn-motion-patterns.py	Fri May 25 18:15:18 2018 -0400
+++ b/scripts/learn-motion-patterns.py	Sun May 27 23:22:48 2018 -0400
@@ -1,4 +1,4 @@
-#! /usr/bin/env python
+#! /usr/bin/env python3
 
 import sys, argparse
 
@@ -61,7 +61,7 @@
     initialPrototypes = storage.loadPrototypesFromSqlite(args.inputPrototypeDatabaseFilename, True)
     trajectories = [p.getMovingObject().getPositions().asArray().T for p in initialPrototypes]+trajectories
     if len(initialPrototypes) > 0:
-        initialPrototypeIndices = range(len(initialPrototypes))
+        initialPrototypeIndices = list(range(len(initialPrototypes)))
     else:
         initialPrototypeIndices = None
 else: