commit 619fb4dcaed62aac6baaf207b283ff3e9c92851d
parent 129288b5bf1a57616a7fa325c958a7892ebcbf0d
Author: Steven Atkinson <[email protected]>
Date: Sun, 15 Sep 2024 17:41:51 -0700
Remove `__all__`s from package (#461)
Remove __all__s from package
Diffstat:
9 files changed, 6 insertions(+), 25 deletions(-)
diff --git a/nam/models/losses.py b/nam/models/losses.py
@@ -11,8 +11,6 @@ from typing import Optional
import torch
from auraloss.freq import MultiResolutionSTFTLoss
-___all__ = ["apply_pre_emphasis_filter", "esr", "multi_resolution_stft_loss"]
-
def apply_pre_emphasis_filter(x: torch.Tensor, coef: float) -> torch.Tensor:
"""
diff --git a/nam/models/metadata.py b/nam/models/metadata.py
@@ -7,12 +7,10 @@ Metadata about models
"""
from enum import Enum
-from typing import List, Optional
+from typing import Optional
from pydantic import BaseModel
-__all__ = ["GearType", "ToneType", "Date", "UserMetadata"]
-
# Note: if you change this enum, you need to update the options in easy_colab.ipynb!
class GearType(Enum):
diff --git a/nam/train/__init__.py b/nam/train/__init__.py
@@ -6,9 +6,7 @@
Code for standardized training with NAM
"""
-__all__ = ["colab", "core", "gui", "metadata"]
-
-from . import colab
-from . import core
-from . import gui
-from . import metadata
+from . import colab # noqa F401
+from . import core # noqa F401
+from . import gui # noqa F401
+from . import metadata # noqa F401
diff --git a/nam/train/_names.py b/nam/train/_names.py
@@ -6,8 +6,6 @@ from typing import NamedTuple, Optional, Set
from ._version import PROTEUS_VERSION, Version
-__all__ = ["INPUT_BASENAMES", "LATEST_VERSION", "VersionAndName"]
-
class VersionAndName(NamedTuple):
version: Version
diff --git a/nam/train/colab.py b/nam/train/colab.py
@@ -15,9 +15,6 @@ from ._version import PROTEUS_VERSION, Version
from .core import TrainOutput, train
from .metadata import TRAINING_KEY
-__all__ = ["run"]
-
-
_BUGGY_INPUT_BASENAMES = {
# 1.1.0 has the spikes at the wrong spots.
"v1_1_0.wav"
diff --git a/nam/train/gui/__init__.py b/nam/train/gui/__init__.py
@@ -41,8 +41,6 @@ except ImportError:
_install_is_valid = False
_HAVE_ACCELERATOR = False
-__all__ = ["run"]
-
if _HAVE_ACCELERATOR:
_DEFAULT_NUM_EPOCHS = 100
_DEFAULT_BATCH_SIZE = 16
diff --git a/nam/train/gui/_resources/__init__.py b/nam/train/gui/_resources/__init__.py
@@ -2,6 +2,4 @@
# Created Date: Tuesday May 14th 2024
# Author: Steven Atkinson ([email protected])
-from . import settings
-
-__all__ = ["settings"]
+from . import settings # noqa F401
diff --git a/nam/train/gui/_resources/settings.py b/nam/train/gui/_resources/settings.py
@@ -8,8 +8,6 @@ from functools import partial
from pathlib import Path
from typing import Optional
-__all__ = ["PathKey", "get_last_path", "set_last_path"]
-
_THIS_DIR = Path(__file__).parent.resolve()
_SETTINGS_JSON_PATH = Path(_THIS_DIR, "settings.json")
_LAST_PATHS_KEY = "last_paths"
diff --git a/nam/util.py b/nam/util.py
@@ -9,8 +9,6 @@ Helpful utilities
import warnings
from datetime import datetime
-__all__ = ["filter_warnings", "timestamp"]
-
def timestamp() -> str:
t = datetime.now()
