diff --git a/prototorch/core/components.py b/prototorch/core/components.py index 5d474a7..c709bf3 100644 --- a/prototorch/core/components.py +++ b/prototorch/core/components.py @@ -6,7 +6,8 @@ from typing import Union import torch from torch.nn.parameter import Parameter -from ..utils import parse_distribution +from prototorch.utils import parse_distribution + from .initializers import ( AbstractClassAwareCompInitializer, AbstractComponentsInitializer, diff --git a/prototorch/core/initializers.py b/prototorch/core/initializers.py index 4518bc7..909628a 100644 --- a/prototorch/core/initializers.py +++ b/prototorch/core/initializers.py @@ -11,7 +11,7 @@ from typing import ( import torch -from ..utils import parse_data_arg, parse_distribution +from prototorch.utils import parse_data_arg, parse_distribution # Components diff --git a/prototorch/core/losses.py b/prototorch/core/losses.py index f413de5..c0977d3 100644 --- a/prototorch/core/losses.py +++ b/prototorch/core/losses.py @@ -2,7 +2,7 @@ import torch -from ..nn.activations import get_activation +from prototorch.nn.activations import get_activation # Helpers