From 099dd15291ee612aa31311d79a9c47c6d9a6dc96 Mon Sep 17 00:00:00 2001
From: Zack Witten <zswitten@gmail.com>
Date: Thu, 2 Aug 2018 14:03:17 -0700
Subject: [PATCH] Cast as int

---
 gen_anchors.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/gen_anchors.py b/gen_anchors.py
index c6e8e2cce..3ff4e96cb 100644
--- a/gen_anchors.py
+++ b/gen_anchors.py
@@ -102,7 +102,7 @@ def run_kmeans(ann_dims, anchor_num):
 
 def main(argv):
     config_path = args.conf
-    num_anchors = args.anchors
+    num_anchors = int(args.anchors)
 
     with open(config_path) as config_buffer:
         config = json.loads(config_buffer.read())