Fix subpackage installation by adding __init__.py files and updating package discovery

Author: Nikolas Borrel-Jensen

deeponet_acoustics/datahandlers/__init__.py

@@ -290,7 +290,7 @@ def __init__(self, settings: dict, num_srcs: int = -1):
 
         self.data_path = settings["test_data_dir"]
         self.model_dir = settings["model_dir"]
-        self.tmax = settings["tmax"]
+        self.tmax = settings["tmax"] if "tmax" in settings else float("inf")
 
         self.snap_to_grid = settings["snap_to_grid"]
         self.write_full_wave_field = settings["write_full_wave_field"]

deeponet_acoustics/end2end/__init__.py

@@ -38,7 +38,7 @@ def __init__(self, settings, input_dir=None, output_dir=None):
             settings, input_dir=input_dir, output_dir=output_dir
         )
 
-        self.tmax = settings["tmax"]
+        self.tmax = settings["tmax"] if "tmax" in settings else float("inf")
         self.f0_feat = settings["f0_feat"] if "f0_feat" in settings else []
         self.normalize_data = (
             settings["normalize_data"] if "normalize_data" in settings else True

deeponet_acoustics/models/__init__.py

@@ -48,8 +48,8 @@ test = [
    lint.select = ["E", "F", "I"]  # Error, pyflakes, isort rules
    lint.ignore = ["E501", "E731"]  # Ignore: line too long, lambda assignments
 
-[tool.setuptools]
-packages = ["deeponet_acoustics"] 
+[tool.setuptools.packages.find]
+exclude = ["deeponet_acoustics.scripts*"]
 
 [project.scripts]
 deeponet-train = "deeponet_acoustics.main_train:main"