Increased code DRYness

Author: Cameron Mattson

src/comparison_tools/PairwiseCompareManager.py

@@ -1,8 +1,8 @@
 from typing import Optional
 
 import pandas as pd
-from comparators.Comparator import Comparator
 
+from comparators.Comparator import Comparator
 from comparison_tools.PairwiseCompare import PairwiseCompare
 
 
@@ -61,27 +61,22 @@ def __init__(
         self._comparator = _comparator
 
         if len(_different_columns) >= 2 and not _same_columns:
-            pairwise_compare_obj = PairwiseCompare(
-                _df=_df,
-                _comparator=_comparator,
-                _antehoc_group_cols=_different_columns[:1],
-                _posthoc_group_cols=_different_columns[1:],
-                _feat_cols=_feat_cols,
-                _drop_cols=_drop_cols,
-            )
+            _same_columns = _different_columns[:1].copy()
+            _different_columns = _different_columns[1:]
+
+        pairwise_compare_obj = PairwiseCompare(
+            _df=_df,
+            _comparator=_comparator,
+            _antehoc_group_cols=_same_columns,
+            _posthoc_group_cols=_different_columns,
+            _feat_cols=_feat_cols,
+            _drop_cols=_drop_cols,
+        )
 
+        if len(_different_columns) >= 2 and not _same_columns:
             pairwise_compare_obj.inter_comparisons()
 
         elif len(_different_columns) >= 1 and len(_same_columns) >= 1:
-            pairwise_compare_obj = PairwiseCompare(
-                _df=_df,
-                _comparator=_comparator,
-                _antehoc_group_cols=_same_columns,
-                _posthoc_group_cols=_different_columns,
-                _feat_cols=_feat_cols,
-                _drop_cols=_drop_cols,
-            )
-
             pairwise_compare_obj.intra_comparisons()
 
     def __call__(self):