sonusai 1.0.6__tar.gz → 1.0.7__tar.gz

{sonusai-1.0.6 → sonusai-1.0.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: sonusai
-Version: 1.0.6
+Version: 1.0.7
 Summary: Framework for building deep neural network models for sound, speech, and voice AI
 Home-page: https://aaware.com
 License: GPL-3.0-only

{sonusai-1.0.6 → sonusai-1.0.7}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "sonusai"
-version = "1.0.6"
+version = "1.0.7"
 description = "Framework for building deep neural network models for sound, speech, and voice AI"
 authors = ["Chris Eddington <chris@aaware.com>", "Jason Calderwood <jason@aaware.com>"]
 maintainers = ["Chris Eddington <chris@aaware.com>", "Jason Calderwood <jason@aaware.com>"]

{sonusai-1.0.6 → sonusai-1.0.7}/sonusai/__init__.py

@@ -16,9 +16,9 @@ commands_doc = """
    genmix                       Generate mixture and truth data
    genmixdb                     Generate a mixture database
    lsdb                         List information about a mixture database
+   metrics_summary              Summarize generated metrics in a mixture database
    mkwav                        Make WAV files from a mixture database
    onnx_predict                 Run ONNX predict on a trained model
-   summarize_metric_spenh       Summarize speech enhancement and analysis results
    vars                         List custom SonusAI variables
 """
 

{sonusai-1.0.6 → sonusai-1.0.7}/sonusai/metrics_summary.py

@@ -90,8 +90,8 @@ def _process_mixture(
 def main() -> None:
     from docopt import docopt
 
-    from . import __version__ as sai_version
-    from .utils.docstring import trim_docstring
+    from sonusai import __version__ as sai_version
+    from sonusai.utils import trim_docstring
 
     args = docopt(trim_docstring(__doc__), version=sai_version, options_first=True)
 
@@ -107,14 +107,14 @@ def main() -> None:
 
     import psutil
 
-    from . import create_file_handler
-    from . import initial_log_messages
-    from . import logger
-    from . import update_console_handler
-    from .mixture.mixdb import MixtureDatabase
-    from .utils.create_timestamp import create_timestamp
-    from .utils.parallel import par_track
-    from .utils.parallel import track
+    from sonusai import create_file_handler
+    from sonusai import initial_log_messages
+    from sonusai import logger
+    from sonusai import update_console_handler
+    from sonusai.mixture import MixtureDatabase
+    from sonusai.utils import create_timestamp
+    from sonusai.utils import par_track
+    from sonusai.utils import track
 
     mixdb = MixtureDatabase(location)
     print(f"Found SonusAI mixture database with {mixdb.num_mixtures} mixtures.")

{sonusai-1.0.6 → sonusai-1.0.7}/sonusai/mixture/mixdb.py

@@ -2180,7 +2180,7 @@ class MixtureDatabase:
 
             # Check for metrics dependencies and add them even if not explicitly requested.
             if metric.startswith("mxwer"):
-                dependencies = ("mxasr." + metric[6:], "tasr." + metric[6:])
+                dependencies = ("mxasr." + metric[6:], "sasr." + metric[6:])
                 for dependency in dependencies:
                     result[dependency] = calc(dependency)
 

sonusai-1.0.6/sonusai/summarize_metric_spenh.py

@@ -1,65 +0,0 @@
-"""sonusai summarize_metric_spenh
-
-usage: summarize_metric_spenh [-hr] [-s SORT] LOC
-
-options:
-    -h, --help
-    -s SORT, --sort SORT            Sort by SORT column. [default: MIXID]
-    -r, --reverse                   Reverse sort order.
-
-Summarize speech enhancement metrics results using data generated by SonusAI calc_metric_spenh.
-
-Inputs:
-    LOC         A SonusAI calc_metric_spenh results directory.
-
-"""
-
-
-def summarize_metric_spenh(location: str, by: str = "MIXID", reverse: bool = False) -> str:
-    import glob
-
-    import pandas as pd
-
-    files = sorted(glob.glob(location + "/*_metric_spenh.txt"))
-    need_header = True
-    header = ["MIXID"]
-    data = []
-    for file in files:
-        with open(file) as f:
-            for i, line in enumerate(f):
-                if i == 1 and need_header:
-                    need_header = False
-                    header.extend(line.strip().split())
-                elif i == 2:
-                    data.append(line.strip().split())
-                    break
-
-    df = pd.DataFrame(data, columns=header)  # pyright: ignore [reportArgumentType]
-    df[header[0:-2]] = df[header[0:-2]].apply(pd.to_numeric, errors="coerce")
-    return df.sort_values(by=by, ascending=not reverse).to_string(index=False)
-
-
-def main():
-    from docopt import docopt
-
-    from sonusai import __version__ as sai_version
-    from sonusai.utils import trim_docstring
-
-    args = docopt(trim_docstring(__doc__), version=sai_version, options_first=True)
-
-    by = args["--sort"]
-    reverse = args["--reverse"]
-    location = args["LOC"]
-
-    print(summarize_metric_spenh(location, by, reverse))
-
-
-if __name__ == "__main__":
-    from sonusai import exception_handler
-    from sonusai.utils import register_keyboard_interrupt
-
-    register_keyboard_interrupt()
-    try:
-        main()
-    except Exception as e:
-        exception_handler(e)