make it work with python3

ProjectQ-Framework · Jul 23, 2017 · ba224de · ba224de
1 parent 978c365
commit ba224de
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/fermilib/utils/_operator_utils.py b/src/fermilib/utils/_operator_utils.py
@@ -20,7 +20,7 @@
 
 from fermilib.config import *
 from fermilib.ops import *
-from itertools import *
+from future.builtins.iterators import map, zip
 
 from projectq.ops import QubitOperator
 
@@ -153,8 +153,8 @@ def save_operator(operator, file_name=None, data_directory=None):
 
     tm = operator.terms
     with open(file_path, 'wb') as f:
-        marshal.dump((operator_type, dict(izip(tm.iterkeys(),
-                                          imap(complex, tm.itervalues())))), f)
+        marshal.dump((operator_type, dict(zip(tm.keys(),
+                                          map(complex, tm.values())))), f)
         f.close()