Skip to content

Commit dbd5620

Browse files
committed
#5: Fix after rebase
1 parent ed74b95 commit dbd5620

File tree

2 files changed

+8
-6
lines changed

2 files changed

+8
-6
lines changed

detection/detect_slow_nodes.py

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@ class SlowNodeDetector:
3333
"""
3434

3535
def __init__(
36-
self, path, sensors, num_nodes, pct, spn, rpn, plot_rank_breakdowns):
36+
self, path, sensors, num_nodes, pct, spn, rpn, plot_rank_breakdowns, use_clstr):
3737
# Create empty dicts for storing data
3838
self.__rank_times = {}
3939
self.__rank_breakdowns = {}
@@ -52,7 +52,7 @@ def __init__(
5252
self.__temperature_analysis_available = True if self.__sensors_output_file is not None else False
5353
self.__plot_rank_breakdowns = plot_rank_breakdowns
5454
self.__num_ranks = 0
55-
self.__use_clustering = use_clustering
55+
self.__use_clustering = use_clstr
5656

5757
# Initialize outliers
5858
self.__slow_ranks = {}
@@ -663,10 +663,11 @@ def main():
663663
path=filepath,
664664
sensors=sensors_filepath,
665665
num_nodes=args.num_nodes,
666-
threshold_percentage=args.threshold,
667-
sockets_per_node=args.spn,
668-
ranks_per_node=args.rpn,
669-
plot_rank_breakdowns=args.plot_all_ranks)
666+
pct=args.threshold,
667+
spn=args.spn,
668+
rpn=args.rpn,
669+
plot_rank_breakdowns=args.plot_all_ranks,
670+
use_clstr=args.use_clustering)
670671

671672
slowNodeDetector.detect()
672673
slowNodeDetector.createHostfile()

tests/unit/detection/test_slow_node_detector.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,7 @@ def setUp(self):
3131
spn=self.spn,
3232
rpn=self.rpn,
3333
plot_rank_breakdowns=False,
34+
use_clstr=False
3435
)
3536

3637
# Run detection

0 commit comments

Comments
 (0)