Fix pretrained paths in wheel??

Author: cwindolf

.github/workflows/ci.yml

@@ -20,11 +20,13 @@ jobs:
 
       - name: Checkout
         uses: actions/checkout@v4
+        with:
+          path: dartsort
 
       - name: Setup Mamba
         uses: mamba-org/setup-micromamba@v1
         with:
-          environment-file: environment.yml
+          environment-file: dartsort/environment.yml
           init-shell: bash
 
       - name: Install Torch (CPU)
@@ -33,11 +35,11 @@ jobs:
         
       - name: Install Pip Requirements
         run: |
-          python -m pip install -r requirements.txt
+          python -m pip install -r dartsort/requirements.txt
 
       - name: Install DARTsort
         run: |
-          python -m pip install .
+          python -m pip install dartsort/
 
       - name: Get DREDge
         uses: actions/checkout@v4
@@ -51,4 +53,4 @@ jobs:
 
       - name: Pytest
         run: |
-          python -m pytest tests/*
+          python -m pytest dartsort/tests/*

pretrained/detect_np1.pt renamed to src/dartsort/pretrained/detect_np1.pt

@@ -1,16 +1,20 @@
-from pathlib import Path
-
 import torch
 from dartsort.util.waveform_util import (get_channels_in_probe,
                                          set_channels_in_probe)
 from torch import nn
 
 from .transform_base import BaseWaveformDenoiser
 
-default_pretrained_path = (
-    Path(__file__).parent.parent.parent.parent
-    / "pretrained/single_chan_denoiser.pt"
-)
+try:
+    from importlib.resources import files
+except ImportError:
+    try:
+        from importlib_resources import files
+    except ImportError:
+        raise ValueError("Need python>=3.10 or pip install importlib_resources.")
+
+default_pretrained_path = files("dartsort.pretrained")
+default_pretrained_path = default_pretrained_path.joinpath("single_chan_denoiser.pt")
 
 
 class SingleChannelWaveformDenoiser(BaseWaveformDenoiser):