Skip to content

Commit

Permalink
chore: remove unnecessary docstrings
Browse files Browse the repository at this point in the history
  • Loading branch information
IamGianluca committed Apr 6, 2024
1 parent 643725e commit 4d5c2ff
Showing 1 changed file with 0 additions and 6 deletions.
6 changes: 0 additions & 6 deletions blazingai/metrics.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,16 +27,10 @@ def __init__(
self.preds = torchmetrics.CatMetric()

def update(self, preds: torch.Tensor, target: torch.Tensor) -> None: # type: ignore
"""Update state with predictions and targets.
Args:
preds: Predictions from model
target: Ground truth values
"""
self.target.update(target)
self.preds.update(preds)

def compute(self) -> torch.Tensor:
"""Computes mean squared error over state."""
preds = self.preds.compute()
target = self.target.compute()

Expand Down

0 comments on commit 4d5c2ff

Please sign in to comment.