[Parallel(n_jobs=20)]: Using backend LokyBackend with 20 concurrent workers.
[Parallel(n_jobs=20)]: Done  32 tasks      | elapsed:  7.8min
[Parallel(n_jobs=20)]: Done 113 out of 126 | elapsed: 23.2min remaining:  2.7min
[Parallel(n_jobs=20)]: Done 126 out of 126 | elapsed: 26.3min finished
Traceback (most recent call last):
  File "/home/tanpv/Progs/claude/HW/src/exp3_week_s1b.py", line 303, in <module>
    main()
  File "/home/tanpv/Progs/claude/HW/src/exp3_week_s1b.py", line 277, in main
    p1 = pick_p1(po, oo)
         ^^^^^^^^^^^^^^^
  File "/home/tanpv/Progs/claude/HW/src/exp3_week_s1.py", line 157, in pick_p1
    if p[obs.astype(bool)].size == 0:
       ~^^^^^^^^^^^^^^^^^^
IndexError: boolean index did not match indexed array along axis 0; size of axis is 34938 but size of corresponding boolean axis is 34944