mypy8

Author: SeBorgey

autointent/modules/scoring/_gcn/gcn_model.py

@@ -95,7 +95,7 @@ def set_correlation_matrix(self, train_labels: torch.Tensor) -> None:
         corr_matrix = self.create_correlation_matrix(
             train_labels, self.num_classes, self.p_reweight, self.tau_threshold
         )
-        self.correlation_matrix.data.copy_(corr_matrix)
+        self.correlation_matrix.copy_(corr_matrix)
 
     def forward(self, bert_features: torch.Tensor, label_embeddings: torch.Tensor) -> torch.Tensor:
         classifiers = label_embeddings
@@ -134,4 +134,4 @@ def load(cls, path: Path, device: str | None = None) -> Self:
 
         instance = instance.to(device)
         instance.eval()
-        return instance
+        return instance