21601: Move --short-tests-only check to arvados_version

[arvados.git] / services / dockercleaner / setup.py

diff --git a/services/dockercleaner/setup.py b/services/dockercleaner/setup.py
index 2b386c70b47aa2c925b87aedff14838297b88315..565fdcd1ea3c582057bf70d9fdf8633c3c45dd94 100644 (file)
--- a/services/dockercleaner/setup.py
+++ b/services/dockercleaner/setup.py
@@ -15,11 +15,7 @@ README = os.path.join(SETUP_DIR, 'README.rst')
 
 import arvados_version
 version = arvados_version.get_version(SETUP_DIR, "arvados_docker")
-
-short_tests_only = False
-if '--short-tests-only' in sys.argv:
-    short_tests_only = True
-    sys.argv.remove('--short-tests-only')
+short_tests_only = arvados_version.short_tests_only()
 
 setup(name="arvados-docker-cleaner",
       version=version,