[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

torch_points3d/metrics/base_tracker.py

@@ -1,4 +1,4 @@
-from typing import Dict, Any, Callable, Optional
+from typing import Any, Dict, Optional
 import torch
 from torch import nn
 from torchmetrics import AverageMeter

torch_points3d/metrics/segmentation/metrics.py

@@ -1,5 +1,5 @@
 import torch
-from typing import Dict, Optional, Tuple, Any, Union
+from typing import Optional, Tuple, Union
 
 
 def compute_average_intersection_union(confusion_matrix: torch.Tensor, missing_as_one: bool = False) -> torch.Tensor:

torch_points3d/metrics/segmentation/segmentation_tracker.py

@@ -1,5 +1,5 @@
 import torch
-from typing import Dict, Optional, Tuple, Any, Union
+from typing import Any, Dict
 from torchmetrics import ConfusionMatrix
 from torchmetrics import Metric
 

torch_points3d/models/base_model.py

@@ -23,10 +23,9 @@ def forward(self) -> Optional[torch.Tensor]:
     def compute_loss(self):
         raise (NotImplementedError("get_losses needs to be defined!"))
 
-    def get_losses(self) -> Optional[Dict["str", torch.Tensor]]:        
+    def get_losses(self) -> Optional[Dict["str", torch.Tensor]]:
         return self._losses
 
-
     def get_outputs(self) -> Dict[str, Optional[torch.Tensor]]:
         """
         return the outputs to track for the metrics