Skip to content

Commit c78bfca

Browse files
committed
ft-analyzer statistical_model.py read csv file completely in chunks
expect merges. They can't be easily refactored like that.
1 parent ee5fcba commit c78bfca

2 files changed

Lines changed: 134 additions & 135 deletions

File tree

tools/ft-analyzer/ftanalyzer/models/precise_model.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -136,9 +136,8 @@ def validate_precise(
136136
if segments is None:
137137
segments = [None]
138138

139-
if not hasattr(self, "_flows"):
140-
self._flows = self._load_flows_df()
141-
self._ref = self._load_ref_df()
139+
flows_df = self._load_flows_df()
140+
ref_df = self._load_ref_df()
142141

143142
# precise model compares IP addresses, so the same format is required
144143
if self._flows_ip_addresses_converted != self._ref_ip_addresses_converted:
@@ -147,7 +146,8 @@ def validate_precise(
147146
all_flow_masks = []
148147
for segment in segments:
149148
self._report.add_segment(segment)
150-
flows, refs, mask_flow = self._filter_segment(segment)
149+
flows, mask_flow = self._filter_segment(segment, flows_df)
150+
refs, _ = self._filter_segment(segment, ref_df)
151151
all_flow_masks.append(mask_flow)
152152

153153
# perform outer join thank to which we can easily identify flows

0 commit comments

Comments
 (0)