Skip to content

Commit

Permalink
Code quality fixes
Browse files Browse the repository at this point in the history
  • Loading branch information
georgeyiasemis committed Apr 18, 2024
1 parent 83ab7fb commit 49577ad
Showing 1 changed file with 2 additions and 4 deletions.
6 changes: 2 additions & 4 deletions direct/ssl/ssl.py
Original file line number Diff line number Diff line change
Expand Up @@ -205,8 +205,7 @@ def _gaussian_split(
if self.keep_acs:
if acs_mask is None:
raise ValueError("`keep_acs` is set to True but not received an input for `acs_mask`.")
else:
mask = mask & (~acs_mask)
mask = mask & (~acs_mask)

with temp_seed(self.rng, seed):
if seed is None:
Expand Down Expand Up @@ -278,8 +277,7 @@ def _uniform_split(
if self.keep_acs:
if acs_mask is None:
raise ValueError("`keep_acs` is set to True but not received an input for `acs_mask`.")
else:
mask = mask & (~acs_mask)
mask = mask & (~acs_mask)

temp_mask = mask.cpu().clone()

Expand Down

0 comments on commit 49577ad

Please sign in to comment.