printing some lens

global_func/stratification_function.py

@@ -25,10 +25,13 @@ def stratification_function(portfolio: pd.DataFrame, lineup_target: int, exclude
             closest_idx = (portfolio[sorting_choice] - target_sim).abs().idxmin()
             if closest_idx not in selected_indices:  # Avoid duplicates
                 selected_indices.append(closest_idx)
+        print(len(selected_indices))
         if len(selected_indices) > lineup_target:
             selected_indices = selected_indices[:lineup_target]
+            print(len(selected_indices))
             range_var = 10
         elif len(selected_indices) == lineup_target:
+            print(len(selected_indices))
             range_var = 10
         else:
             lineup_target += 5 * range_var