summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--test_trove_upgrades.py2
-rw-r--r--util.py1
2 files changed, 2 insertions, 1 deletions
diff --git a/test_trove_upgrades.py b/test_trove_upgrades.py
index b564709..a6d940b 100644
--- a/test_trove_upgrades.py
+++ b/test_trove_upgrades.py
@@ -331,7 +331,7 @@ class SystemTestBranch(object):
Uses `morph edit` and `git reset` to achieve this.
'''
chunk_repo_uri, original_ref = self.get_repo_and_ref_for_chunk(
- branch, stratum, chunk)
+ stratum, chunk)
with set_directory(self.morphs_dir):
run_morph(['edit', system, stratum, chunk])
diff --git a/util.py b/util.py
index 8974674..15c9c72 100644
--- a/util.py
+++ b/util.py
@@ -17,6 +17,7 @@
import cliapp
import contextlib
+import os
import sys
import urlparse