human
Solve the following python programming problem given the following code:
def test_where() : 
    def fun(x, y) : 
        b = np.where (C, x, y) 
        return to_scalar (b) 
C = (npr.randn (4, 5) > 0) 
    A = npr.randn (4, 5) 
    B = npr.randn (4, 5) 
    d_fun = (lambda a, b : to_scalar (grad (fun) (a, b))) 
    check_grads (fun, A, B) 
    check_grads (d_fun, A, d_fun) 

    
    

    Fix the buggy line: check_grads (d_fun, A, d_fun)
assistant
#Here is Python code to fix the buggy line: check_grads (d_fun, A, d_fun)
.

check_grads (d_fun, A, B)