fix

nndet/losses/classification.py

@@ -25,9 +25,6 @@ from nndet.losses.base import reduction_helper
 from nndet.utils import make_onehot_batch
 
 
-__all__ = ["focal_loss_with_logits", "FocalLossWithLogits"]
-
-
 def one_hot_smooth(data,
                    num_classes: int,
                    smoothing: float = 0.0,

nndet/losses/regression.py

-"""
-Copyright 2020 Division of Medical Image Computing, German Cancer Research Center (DKFZ), Heidelberg, Germany
-
-Licensed under the Apache License, Version 2.0 (the "License");
-you may not use this file except in compliance with the License.
-You may obtain a copy of the License at
-
-   http://www.apache.org/licenses/LICENSE-2.0
-
-Unless required by applicable law or agreed to in writing, software
-distributed under the License is distributed on an "AS IS" BASIS,
-WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-See the License for the specific language governing permissions and
-limitations under the License.
-"""
-
 from typing import Optional
 
 import torch

nndet/losses/segmentation.py

@@ -20,8 +20,6 @@ import torch.nn as nn
 from torch import Tensor
 from typing import Callable
 
-__all__ = ["SoftDiceLoss"]
-
 
 def one_hot_smooth_batch(data, num_classes: int, smoothing: float = 0.0):
     shape = data.shape