Part of Trump's ban on diversity training is blocked by a California federal judge