sonusai 1.0.6__py3-none-any.whl → 1.0.7__py3-none-any.whl

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/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/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.dist-info → sonusai-1.0.7.dist-info}/METADATA

@@ -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.dist-info → sonusai-1.0.7.dist-info}/RECORD

@@ -1,4 +1,4 @@
-sonusai/__init__.py,sha256=_lBs_XangUYtUcr7JAt55hk1AczFIAgySJfkL-Pt51U,3710
+sonusai/__init__.py,sha256=XSjulPEFTHCTNAte0z0Ie0e-3w9uVX5j5mR9QfE_r40,3710
 sonusai/aawscd_probwrite.py,sha256=QZLMQrmPr3OjZ06buyYDwlnk9YPCpyr4KHkBjPsiqjU,3700
 sonusai/audiofe.py,sha256=0DNpntK0WpzNZeyHX8_wC-pDtrgvLkJZFPXz-PspdrY,19448
 sonusai/calc_metric_spenh.py,sha256=0Md6hRFUH9lGnsvoydqne99O7Gi0ieG9vMU_1PnASBg,50019
@@ -44,7 +44,7 @@ sonusai/metrics/class_summary.py,sha256=mQbMxQ8EtFIN7S2h7A4Dk0X4XF_CIxKk3W8zZMmp
 sonusai/metrics/confusion_matrix_summary.py,sha256=lhd8TyHVMC03khX85h_D75XElmawx56KkqpX3X2O2gQ,3133
 sonusai/metrics/one_hot.py,sha256=aKc-xYd4zWIjbmoQikIcQ6BJB1k-68XKTg8eJCacHTU,13906
 sonusai/metrics/snr_summary.py,sha256=qKHctpmvGeu2cmjTG7iQPX1lvVUEtEnCIKwUGu6VrEQ,5773
-sonusai/metrics_summary.py,sha256=qfQ_NAW4ilgyISRHxSuewTK-u_eo4bkjOjLNX-qd1z4,12197
+sonusai/metrics_summary.py,sha256=pQ5kNwweQM6LdP6TpJE7kV3Z9BYUd96IfEjnZLfSpQI,12211
 sonusai/mixture/__init__.py,sha256=_vepE2uhAGKHIujPWxfGDeaWHP5yKLf5BjXkU9ZereA,1258
 sonusai/mixture/audio.py,sha256=JyrVtVPLH3aTXFgyl446f5uVHxlFRa4aBaSPYaMdg80,5814
 sonusai/mixture/class_balancing.py,sha256=lubicVCzxs4TMh2dZSsuIffkLkk1gmwjmwtrtQ27BVQ,3638
@@ -59,7 +59,7 @@ sonusai/mixture/helpers.py,sha256=dmyHwf1C5dZjYOd11kVV16KI33CaM-dU_fyaxOrrKt8,11
 sonusai/mixture/ir_delay.py,sha256=aiC23HMWQ08-v5wORgMx1_DOJSdh4kunULqiQ-SGuMo,2026
 sonusai/mixture/ir_effects.py,sha256=PqiqD4PS42-7kD6ESnsZi2a3tnKCFa4E0xqUujRBvGg,2152
 sonusai/mixture/log_duration_and_sizes.py,sha256=3ekS27IMKlnxIkQAmprzmBnzHOpRjZh3d7maL2VqWQU,927
-sonusai/mixture/mixdb.py,sha256=QIpbpS-J0ACNUFoMwZjiylsDgpfkUehLxSq79wcLgzk,86305
+sonusai/mixture/mixdb.py,sha256=zGFagqRIV9uX2QiP795lyN29AarGnZgeKTdUIBcuyfY,86305
 sonusai/mixture/pad_audio.py,sha256=KNxVQAejA0hblLOnMJgLS6lFaeE0n3tWQ5rclaHBnIY,1015
 sonusai/mixture/resample.py,sha256=jXqH6FrZ0mlhQ07XqPx88TT9elu3HHVLw7Q0a7Lh5M4,221
 sonusai/mixture/sox_effects.py,sha256=tndS9qrh3eJOTUPrufyWHCt3UqjbPuh81I4Lo4MNmDg,5328
@@ -88,7 +88,6 @@ sonusai/speech/timit.py,sha256=73T1eOQcA1FfvpCGN8gC4iP56vt4KWcDjLnlkVgdZ9U,4099
 sonusai/speech/types.py,sha256=4eKVPAktpkIrZ2qoVp2iT45zxTVNocQEGT6O_Zlub_w,214
 sonusai/speech/vctk.py,sha256=WInvRRRkZCW6t_NcZAJffJzgCbyetal-j2w0kKX5SDw,1527
 sonusai/speech/voxceleb.py,sha256=Uu1kB1krf8hess1yuvGbYfV_VgYhklEyoz4I7KfrVpw,2658
-sonusai/summarize_metric_spenh.py,sha256=Zha0PZGZ_NznPeHh8rjVIaWR7ac4MEIlijci1uikiT0,1870
 sonusai/utils/__init__.py,sha256=4XB-62-4Dg_XgZ6Ip2fEOBJ0_oQff6jndYl2rlzl6XE,2699
 sonusai/utils/asl_p56.py,sha256=zH82RI7h399ZYFOOiOC35iv_qi0KhXpmG5uItWB8n78,3854
 sonusai/utils/asr.py,sha256=w5xSBpA2wibzsiCWzdSm1m3LwNd54dcJMMLBWN33FJs,2818
@@ -132,7 +131,7 @@ sonusai/utils/tokenized_shell_vars.py,sha256=EDrrAgz5lJ0RBAjLcTJt1MeyjhbNZiqXkym
 sonusai/utils/write_audio.py,sha256=IHzrJoFtFcea_J6wo6QSiojRkgnNOzAEcg-z0rFV7nU,810
 sonusai/utils/yes_or_no.py,sha256=0h1okjXmDNbJp7rZJFR2V-HFU1GJDm3YFTUVmYExkOU,263
 sonusai/vars.py,sha256=m8pdgfR4A6A9TCGf_rok6jPAT5BgrEsYXTSISIh1nrI,1163
-sonusai-1.0.6.dist-info/METADATA,sha256=NeaZn2a1yaOmW9AWfWerYv5_RqTTL_MlQGZJv9XHVmo,2652
-sonusai-1.0.6.dist-info/WHEEL,sha256=RaoafKOydTQ7I_I3JTrPCg6kUmTgtm4BornzOqyEfJ8,88
-sonusai-1.0.6.dist-info/entry_points.txt,sha256=zMNjEphEPO6B3cD1GNpit7z-yA9tUU5-j3W2v-UWstU,92
-sonusai-1.0.6.dist-info/RECORD,,
+sonusai-1.0.7.dist-info/METADATA,sha256=eyPTvxuqs2NwuNq2Rp5JsQOeCCZxol44M6EIWlxW3Ng,2652
+sonusai-1.0.7.dist-info/WHEEL,sha256=RaoafKOydTQ7I_I3JTrPCg6kUmTgtm4BornzOqyEfJ8,88
+sonusai-1.0.7.dist-info/entry_points.txt,sha256=zMNjEphEPO6B3cD1GNpit7z-yA9tUU5-j3W2v-UWstU,92
+sonusai-1.0.7.dist-info/RECORD,,

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)