[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/models/base_model.py

@@ -1,4 +1,4 @@
-from typing import Dict, Union
+from typing import Union
 
 import torch
 import torch.nn as nn
@@ -20,4 +20,4 @@ def forward(self) -> Union[torch.Tensor, None]:
         raise (NotImplementedError("forward needs to be defined!"))
 
     def get_losses(self) -> Union[torch.Tensor, None]:
-        raise (NotImplementedError("get_losses needs to be defined!"))
+        raise (NotImplementedError("get_losses needs to be defined!"))

torch_points3d/models/segmentation/base_model.py

@@ -1,5 +1,5 @@
 from omegaconf import DictConfig
-from typing import Dict, Union
+from typing import Union
 
 import torch
 import torch.nn as nn
@@ -41,4 +41,4 @@ def get_losses(self) -> Union[torch.Tensor, None]:
             return
 
         self.loss = self.criterion(self.output, self.labels)
-        return self.loss
+        return self.loss