fix: fix the imports in base loss

Author: okunator

cellseg_models_pytorch/losses/weighted_base_loss.py

@@ -1,7 +1,7 @@
 import torch
 import torch.nn as nn
 
-from ..utils import filter2D, gaussian_kernel2d
+from cellseg_models_pytorch.utils.convolve import filter2D, gaussian_kernel2d
 
 
 class WeightedBaseLoss(nn.Module):
@@ -13,7 +13,7 @@ def __init__(
         apply_mask: bool = False,
         class_weights: torch.Tensor = None,
         edge_weight: float = None,
-        **kwargs
+        **kwargs,
     ) -> None:
         """Init a base class for weighted cross entropy based losses.
 
@@ -99,7 +99,7 @@ def apply_svls_to_target(
         num_classes: int,
         kernel_size: int = 5,
         sigma: int = 3,
-        **kwargs
+        **kwargs,
     ) -> torch.Tensor:
         """Apply spatially varying label smoothihng to target map.