Skip to content

Commit

Permalink
torch_log_wmse_audio_quality alias updates
Browse files Browse the repository at this point in the history
  • Loading branch information
crlandsc committed Jun 21, 2024
1 parent 9c81e0a commit de47dbc
Show file tree
Hide file tree
Showing 9 changed files with 23 additions and 6 deletions.
2 changes: 1 addition & 1 deletion CHANGELOG.md
Original file line number Diff line number Diff line change
Expand Up @@ -78,7 +78,7 @@ Updated GitHub repo name to appropriate `torch-log-wmse`. `torch-log-wmse-audio-
#### Weighted filter reference bug
Corrected file reference to pkl filter file.

## 0.2.4-0.2.6 (2024-06-21)
## 0.2.4-0.2.7 (2024-06-21)

#### Added `torch_log_wmse_audio_quality` alias
added alias file so imports can either be `torch_log_wmse` or `torch_log_wmse_audio_quality`.
2 changes: 1 addition & 1 deletion setup.cfg
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
[metadata]
name = torch-log-wmse
version = 0.2.6
version = 0.2.7
author = Christopher Landschoot
author_email = [email protected]
license = Apache License 2.0
Expand Down
12 changes: 10 additions & 2 deletions tests/test_metric.py
Original file line number Diff line number Diff line change
@@ -1,11 +1,19 @@
# Add directory to python path if needed
import sys
sys.path.append("/Users/chris/Desktop/Whitebalance/torch-log-wmse")

import unittest
import torch
import numpy as np
import matplotlib.pyplot as plt
from torch_log_wmse import LogWMSE
from torch_log_wmse.utils import calculate_rms
from torch_log_wmse.utils import calculate_rms, convert_decibels_to_amplitude_ratio
from torch_log_wmse.freq_weighting_filter import prepare_impulse_response_fft, HumanHearingSensitivityFilter
from torch_log_wmse.utils import convert_decibels_to_amplitude_ratio

# Test alias package
# from torch_log_wmse_audio_quality import LogWMSE
# from torch_log_wmse_audio_quality.utils import calculate_rms, convert_decibels_to_amplitude_ratio
# from torch_log_wmse_audio_quality.freq_weighting_filter import prepare_impulse_response_fft, HumanHearingSensitivityFilter

class TestLogWMSELoss(unittest.TestCase):
def setUp(self):
Expand Down
2 changes: 1 addition & 1 deletion torch_log_wmse/__init__.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,3 @@
from .metric import LogWMSE

__version__ = "0.2.6"
__version__ = "0.2.7"
3 changes: 2 additions & 1 deletion torch_log_wmse_audio_quality/__init__.py
Original file line number Diff line number Diff line change
@@ -1 +1,2 @@
from torch_log_wmse import LogWMSE
# Alias for torch_log_wmse_audio_quality
from torch_log_wmse import *
2 changes: 2 additions & 0 deletions torch_log_wmse_audio_quality/constants.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
# Alias for torch_log_wmse_audio_quality
from torch_log_wmse.constants import *
2 changes: 2 additions & 0 deletions torch_log_wmse_audio_quality/freq_weighting_filter.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
# Alias for torch_log_wmse_audio_quality
from torch_log_wmse.freq_weighting_filter import *
2 changes: 2 additions & 0 deletions torch_log_wmse_audio_quality/metric.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
# Alias for torch_log_wmse_audio_quality
from torch_log_wmse.metric import *
2 changes: 2 additions & 0 deletions torch_log_wmse_audio_quality/utils.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
# Alias for torch_log_wmse_audio_quality
from torch_log_wmse.utils import *

0 comments on commit de47dbc

Please sign in to comment.