nam.models._base to nam.models.base

sdatkinson · Nov 25, 2024 · cb5eea5 · cb5eea5
1 parent 6bec7cd
commit cb5eea5
Show file tree

Hide file tree

Showing 7 changed files with 7 additions and 7 deletions.
diff --git a/nam/models/__init__.py b/nam/models/__init__.py
@@ -6,7 +6,7 @@
 NAM's neural networks
 """
 
-from . import _base  # noqa F401
+from . import base  # noqa F401
 from . import exportable  # noqa F401
 from . import losses  # noqa F401
 from .conv_net import ConvNet  # noqa F401

diff --git a/nam/models/_base.py → nam/models/base.py b/nam/models/_base.py → nam/models/base.py
diff --git a/nam/models/conv_net.py b/nam/models/conv_net.py
@@ -19,7 +19,7 @@
 from .. import __version__
 from ..data import wav_to_tensor
 from ._activations import get_activation
-from ._base import BaseNet
+from .base import BaseNet
 from ._names import ACTIVATION_NAME, BATCHNORM_NAME, CONV_NAME
 
 

diff --git a/nam/models/linear.py b/nam/models/linear.py
@@ -11,7 +11,7 @@
 import torch.nn as nn
 
 from .._version import __version__
-from ._base import BaseNet
+from .base import BaseNet
 
 
 class Linear(BaseNet):

diff --git a/nam/models/recurrent.py b/nam/models/recurrent.py
@@ -18,7 +18,7 @@
 import torch
 import torch.nn as nn
 
-from ._base import BaseNet
+from .base import BaseNet
 
 
 class _L(nn.LSTM):

diff --git a/nam/models/wavenet.py b/nam/models/wavenet.py
@@ -18,7 +18,7 @@
 import torch.nn as nn
 
 from ._activations import get_activation
-from ._base import BaseNet
+from .base import BaseNet
 from ._names import ACTIVATION_NAME, CONV_NAME
 
 

diff --git a/tests/test_nam/test_models/test_base.py b/tests/test_nam/test_models/test_base.py
@@ -15,10 +15,10 @@
 import pytest
 import torch
 
-from nam.models import _base
+from nam.models import base
 
 
-class MockBaseNet(_base.BaseNet):
+class MockBaseNet(base.BaseNet):
     def __init__(self, gain: float, *args, **kwargs):
         super().__init__(*args, **kwargs)
         self.gain = gain