add flag for --gpu. related to #631

tira-io · Apr 22, 2024 · 8147c37 · 8147c37
1 parent a9a1ff5
commit 8147c37
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/python-client/Makefile b/python-client/Makefile
@@ -3,7 +3,7 @@
 build-pypi-package: run-tests
 	python3 -m build --sdist .
 	python3 -m build --wheel .
-	twine upload dist/tira-0.0.124-py3-none-any.whl dist/tira-0.0.124.tar.gz
+	twine upload dist/tira-0.0.125-py3-none-any.whl dist/tira-0.0.125.tar.gz
 
 run-tests:
 	docker run -u root --rm -v /var/run/docker.sock:/var/run/docker.sock -v ${PWD}:/app -w /app --entrypoint pytest webis/tira:python-client-dev-0.0.5

diff --git a/python-client/tira/__init__.py b/python-client/tira/__init__.py
@@ -1 +1 @@
-__version__ = "0.0.124"
+__version__ = "0.0.125"
diff --git a/python-client/tira/tira_run.py b/python-client/tira/tira_run.py
@@ -266,7 +266,7 @@ def main(args=None):
 ''')
     gpus = 0
     if args.gpus:
-        gpus = -1 if args.gpus == 'all' else int(gpus)
+        gpus = -1 if args.gpus == 'all' else int(args.gpus)
 
     client.local_execution.run(identifier=args.approach, image=args.image, command=args.command, input_dir=input_dir, output_dir=args.output_directory, dry_run=args.dry_run, allow_network=args.allow_network, input_run=args.input_run, additional_volumes=args.v, evaluate=evaluate, eval_dir=args.evaluation_directory, gpu_count=gpus)