Fix relative import for python3 (#5)

safijari · May 17, 2019 · b89bc79 · b89bc79
1 parent 7f92771
commit b89bc79
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 6 deletions.
diff --git a/aprilgrid/__init__.py b/aprilgrid/__init__.py
@@ -1,4 +1,4 @@
-from tagFamilies import t36h11
+from .tagFamilies import t36h11
 from apriltags_eth import make_default_detector
 from collections import namedtuple
 

diff --git a/aprilgrid/createTargetPDF.py b/aprilgrid/createTargetPDF.py
@@ -10,25 +10,26 @@
 import math
 import numpy as np
 
-from tagFamilies import *
+from .tagFamilies import *
 
-class AprilTagCodes:
+
+class AprilTagCodes(object):
     def __init__(self, chosenTagFamiliy):
         try:
             self.chosenTagFamiliy = chosenTagFamiliy
             self.tagCodes = TagFamilies[chosenTagFamiliy][0]
             self.totalBits = TagFamilies[chosenTagFamiliy][1]
         except:
-            print "[ERROR]: Unknown tag familiy."
+            print("[ERROR]: Unknown tag familiy.")
             sys.exit(0)
 
 #borderBits must be consitent with the variable "blackBorder" in the detector code in file ethz_apriltag2/src/TagFamily.cc
 def generateAprilTag(canvas, position, metricSize, tagSpacing, tagID, tagFamililyData, rotation=2, symmCorners=True, borderBits=2):
     #get the tag code
     try:
         tagCode=tagFamililyData.tagCodes[tagID]
-    except:
-        print "[ERROR]: Requested tag ID of {0} not available in the {1} TagFamiliy".format(tagID, tagFamililyData.chosenTagFamiliy)        
+    except Exception:
+        print("[ERROR]: Requested tag ID of {0} not available in the {1} TagFamiliy".format(tagID, tagFamililyData.chosenTagFamiliy))
 
     #calculate the bit size of the tag
     sqrtBits = (math.sqrt(tagFamililyData.totalBits))