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

[arvados.git] / tools / crunchstat-summary / setup.py

diff --git a/tools/crunchstat-summary/setup.py b/tools/crunchstat-summary/setup.py
index 98be9f27025b4c3c7828626dd7676fddbe845dfd..4bb9df2ba940b6a9c9e1fe445a55b481dad61aa5 100755 (executable)
--- a/tools/crunchstat-summary/setup.py
+++ b/tools/crunchstat-summary/setup.py
@@ -20,11 +20,7 @@ if os.environ.get('ARVADOS_BUILDING_VERSION', False):
 else:
     # On dev releases, arvados-python-client may have a different timestamp
     pysdk_dep = "<={}".format(version)
-
-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='crunchstat_summary',
       version=version,