From aa96f2f6b5127722d8ad6d51da1bc5499eb7334c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Rony=20Lepp=C3=A4nen?= Date: Tue, 26 Nov 2024 14:36:14 +0000 Subject: [PATCH] metric collector revert gpu arg name --- ts/metrics/metric_collector.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ts/metrics/metric_collector.py b/ts/metrics/metric_collector.py index 9032d42e41..6a811d1bcc 100644 --- a/ts/metrics/metric_collector.py +++ b/ts/metrics/metric_collector.py @@ -15,7 +15,7 @@ parser = argparse.ArgumentParser() parser.add_argument( - "--gpus", + "--gpu", action="store", help="number of GPUs", type=int @@ -24,6 +24,6 @@ logging.basicConfig(stream=sys.stdout, format="%(message)s", level=logging.INFO) - system_metrics.collect_all(sys.modules['ts.metrics.system_metrics'], arguments.gpus) + system_metrics.collect_all(sys.modules['ts.metrics.system_metrics'], arguments.gpu) check_process_mem_usage(sys.stdin)