fix botched merge conflict

aholmberg · aholmberg · commit 4fd7c2917d69 · 2016-05-04T13:34:20.000-05:00
diff --git a/benchmarks/base.py b/benchmarks/base.py
@@ -72,7 +72,7 @@
 def setup(options):
     log.info("Using 'cassandra' package from %s", cassandra.__path__)
 
-    cluster = Cluster(hosts, schema_metadata_enabled=False, token_metadata_enabled=False)
+    cluster = Cluster(options.hosts, schema_metadata_enabled=False, token_metadata_enabled=False)
     try:
         session = cluster.connect()
 
@@ -107,8 +107,8 @@ def setup(options):
         cluster.shutdown()
 
 
-def teardown(hosts):
-    cluster = Cluster(hosts, schema_metadata_enabled=False, token_metadata_enabled=False)
+def teardown(options):
+    cluster = Cluster(options.hosts, schema_metadata_enabled=False, token_metadata_enabled=False)
     session = cluster.connect()
     if not options.keep_data:
         session.execute("DROP KEYSPACE " + options.keyspace)
