Suppose that the experimenter still wants $10$ people in the treatment group, but instead of the restriction that the treatment group contain exactly $5$ males and $5$ females, the experimenter only imposes the restriction that it contain at least $2$ males and at least $2$ females. How many different treatment groups are possible? Store your answer in the variable num$\_$treatment$\_$groups as an integer.

Hint: Break the problem into a few cases based on the number of males in the treatment group, and use a for$-$loop to do the appropriate calculation for each case.

import math

import numpy as $np$

from random import shuffle

from scipy.stats import binom

from scipy.stats import norm

num$\_$treatment$\_$groups $=0$

for num$\_$males in range $(2,9)$ :

num$\_$females $=10-$num$\_$males

num$\_$ways$\_$males$=$math.comb$(14,$ num$\_$males$)$

num$\_$ways$\_$females$=$math.comb$(17,$ num$\_$females$)$

num$\_$treatment$\_$groups$+=($num$\_$ways$\_$males$*$num$\_$ways$\_$females$)$

num$\_$males $=9$

num$\_$females $=1$

num$\_$ways$\_$males$=$math.comb$(14,$num$\_$males$)$

num$\_$ways$\_$females$=$math.comb$(17,$ num$\_$females$)$

num$\_$treatment$\_$groups$+=($num$\_$ways$\_$males$*$num$\_$ways$\_$females$)$

num$\_$males $=10$

num$\_$females $=0$

num$\_$ways$\_$males$=$math.comb$(14,$ num$\_$males$)$

num$\_$treatment$\_$groups $+=$ num$\_$ways$\_$males

print $($num$\_$treatment$\_$groups$)$

Assertion error problem where is my code going wrong?