sonusai 1.0.3__tar.gz → 1.0.4__tar.gz

{sonusai-1.0.3 → sonusai-1.0.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: sonusai
-Version: 1.0.3
+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

{sonusai-1.0.3 → sonusai-1.0.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "sonusai"
-version = "1.0.3"
+version = "1.0.4"
 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.3 → sonusai-1.0.4}/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: