sonusai 1.0.2__py3-none-any.whl → 1.0.4__py3-none-any.whl

sonusai/genmixdb.py

@@ -91,6 +91,9 @@ def genmixdb(
     if len([file for file in source_files if file.category == "primary"]) == 0:
         raise RuntimeError("Canceled due to no primary sources")
 
+    if logging:
+        logger.info("Populating source file table")
+
     populate_source_file_table(location, source_files, test)
 
     if logging:
@@ -123,6 +126,9 @@ def genmixdb(
     ir_files = get_ir_files(config, show_progress=show_progress)
     logger.info("")
 
+    if logging:
+        logger.info("Populating impulse response file table")
+
     populate_impulse_response_file_table(location, ir_files, test)
 
     if logging:

sonusai/mixture/effects.py

@@ -308,7 +308,7 @@ def evaluate_sai_random_ir(mixdb: MixtureDatabase, text: str) -> str:
         return m.group(1)
 
     if re.match(rand_pattern, text):
-        return f"ir {randint(0, mixdb.num_ir_files)}"  # noqa: S311
+        return f"ir {randint(0, mixdb.num_ir_files - 1)}"  # noqa: S311
 
     if re.match(rand_range_pattern, text):
         try:

sonusai/mixture/generation.py

@@ -437,6 +437,7 @@ def populate_mixture_table(
     # Populate source table
     if logging:
         logger.info("Populating source table")
+    # TODO: refactor this to not load all sources into list; maybe us UNIQUE table modifier?
     sources: list[tuple[str, int, float, bool, float, float, bool, int]] = []
     for mixture in mixtures:
         for source in mixture.all_sources.values():
@@ -577,7 +578,7 @@ def _initialize_mixture_gains(mixdb: MixtureDatabase, mixture: Mixture, sources_
             # Special case for zeroing out noise data
             mixture.all_sources[category].snr_gain = 0
         elif category != "primary":
-            if sources_energy["primary"] == 0:
+            if sources_energy["primary"] == 0 or sources_energy[category] == 0:
                 # Avoid divide-by-zero
                 mixture.all_sources[category].snr_gain = 1
             else:

{sonusai-1.0.2.dist-info → sonusai-1.0.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: sonusai
-Version: 1.0.2
+Version: 1.0.4
 Summary: Framework for building deep neural network models for sound, speech, and voice AI
 Home-page: https://aaware.com
 License: GPL-3.0-only
@@ -36,6 +36,7 @@ Requires-Dist: pyyaml (>=6.0.2,<7.0.0)
 Requires-Dist: requests (>=2.32.3,<3.0.0)
 Requires-Dist: rich (>=13.9.4,<14.0.0)
 Requires-Dist: samplerate (>=0.2.1,<0.3.0)
+Requires-Dist: scikit-learn (>=1.6.1,<2.0.0)
 Requires-Dist: sh (>=2.2.2,<3.0.0)
 Requires-Dist: soundfile (>=0.12.1,<0.13.0)
 Requires-Dist: sox (>=1.5.0,<2.0.0)

{sonusai-1.0.2.dist-info → sonusai-1.0.4.dist-info}/RECORD

@@ -23,7 +23,7 @@ sonusai/doc.py,sha256=ZgFSSI56oNDb-yC3xi-RHMClMjryR2VrgGyi3ggX8gM,1098
 sonusai/genft.py,sha256=jGjtjQQEuPunROkoDOYZ7gdyZEa09EMCVhpHrkBZ7A0,5571
 sonusai/genmetrics.py,sha256=sbcqbjI4YOJd5_Lzor4Re_TK6GUQ5zJuYbhDux8odI0,6184
 sonusai/genmix.py,sha256=U62GPgejGfnDfRgXUosmnVVWvTV07sg46JQEIew0nPg,5744
-sonusai/genmixdb.py,sha256=-GsH7qoWXlmQSDoCd_BLWhs3oKsvsBgK3c6JrhJOg20,10984
+sonusai/genmixdb.py,sha256=_YPIwBn_yqbE6Nik0j_-3vT0lO35Jemq_xDsA7zoCi4,11132
 sonusai/ir_metric.py,sha256=nxS_mARPSZG5Y0G3L8HysOnkPj4v-RGxAxAVBYe-gJI,19600
 sonusai/lsdb.py,sha256=86t6PpsyardRa6VcSJ-KyU1NiTmlg59VUlcSTptJbn0,5078
 sonusai/main.py,sha256=72feJv5XEVJE_CQatmNIL1VD9ca-Mo0QNDbXxLrHrbQ,2619
@@ -52,9 +52,9 @@ sonusai/mixture/config.py,sha256=2_hEndyRXxyBpGzyBFaDT9REYGoK9Q7HQy8vDqPozus,233
 sonusai/mixture/constants.py,sha256=Kklzhf5DL30yb3TpqRbvRUhcFrEXJ4s2S3D_nw4ARxM,1498
 sonusai/mixture/data_io.py,sha256=DV48sFcP2Qp3NBzvcnlptQOXU3aUEcAeLuh3XOtC5jI,5341
 sonusai/mixture/db_datatypes.py,sha256=VvNtbOgt5WSeSnBoVcNGC5gs_7hX_38pDUPjy5KRbG4,1471
-sonusai/mixture/effects.py,sha256=4FK0Mgo-XL3lMEi3MARIt18AJVirDG2mFHM5BiRZe2I,11780
+sonusai/mixture/effects.py,sha256=ghMO-WiSMQc1CvafD0wkt_DGsM2A6Hi_oZS6j-jeZh8,11784
 sonusai/mixture/feature.py,sha256=7GJvFhfqeqerfjy9Vq9aKt-cecgYblK0IypNNo5hgwY,2285
-sonusai/mixture/generation.py,sha256=j19nxSo8bbPS9BtZfWQnMTRtGxoNPFi_tClT3pdeCUM,32433
+sonusai/mixture/generation.py,sha256=iozYcur9oGFDOgovph-F7mzuaD7ie-2ED4NE7kpAPWY,32559
 sonusai/mixture/helpers.py,sha256=dmyHwf1C5dZjYOd11kVV16KI33CaM-dU_fyaxOrrKt8,11642
 sonusai/mixture/ir_delay.py,sha256=aiC23HMWQ08-v5wORgMx1_DOJSdh4kunULqiQ-SGuMo,2026
 sonusai/mixture/ir_effects.py,sha256=PqiqD4PS42-7kD6ESnsZi2a3tnKCFa4E0xqUujRBvGg,2152
@@ -132,7 +132,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.2.dist-info/METADATA,sha256=WYBGGOpDWynuruSwOPbNdIXJSrAvcf1k_MXmMYWhU_o,2607
-sonusai-1.0.2.dist-info/WHEEL,sha256=RaoafKOydTQ7I_I3JTrPCg6kUmTgtm4BornzOqyEfJ8,88
-sonusai-1.0.2.dist-info/entry_points.txt,sha256=zMNjEphEPO6B3cD1GNpit7z-yA9tUU5-j3W2v-UWstU,92
-sonusai-1.0.2.dist-info/RECORD,,
+sonusai-1.0.4.dist-info/METADATA,sha256=zoOsV6OKVVFihoDmuEyN9mDCLoNPww_3WdB3d-YsJ3Y,2652
+sonusai-1.0.4.dist-info/WHEEL,sha256=RaoafKOydTQ7I_I3JTrPCg6kUmTgtm4BornzOqyEfJ8,88
+sonusai-1.0.4.dist-info/entry_points.txt,sha256=zMNjEphEPO6B3cD1GNpit7z-yA9tUU5-j3W2v-UWstU,92
+sonusai-1.0.4.dist-info/RECORD,,