Use boolean mask to avoid deprecated usage#14

soraxas wants to merge 1 commit intokamenbliznashki:masterkamenbliznashki/normalizing_flows:masterfrom

soraxas:mastersoraxas/normalizing_flows:masterCopy head branch name to clipboard