Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Apr 8, 2024
1 parent 09bfb20 commit ad82066
Showing 1 changed file with 1 addition and 3 deletions.
4 changes: 1 addition & 3 deletions src/multigrate/module/_multivae_torch.py
Original file line number Diff line number Diff line change
Expand Up @@ -455,9 +455,7 @@ def generative(
rs = [self._h_to_x(z, mod) for mod, z in enumerate(zs)]
return {"rs": rs}

def loss(
self, tensors, inference_outputs, generative_outputs, kl_weight: float = 1.0
) -> Tuple[
def loss(self, tensors, inference_outputs, generative_outputs, kl_weight: float = 1.0) -> Tuple[
torch.FloatTensor,
Dict[str, torch.FloatTensor],
torch.FloatTensor,
Expand Down

0 comments on commit ad82066

Please sign in to comment.