inference-only: drop zcodec deps (relative imports) and avoid training/tokenizer imports at package init

codec/__init__.py

@@ -1,2 +1,2 @@
-from .train_patchvae import TrainPatchVAE
-from .train_wavvae import TrainWavVAE
+# codec/__init__.py — minimal inference exports
+from .models import PatchVAE, PatchVAEConfig, WavVAE  # noqa: F401

codec/models/__init__.py

@@ -1,2 +1,7 @@
-from .patchvae.model import PatchVAE, PatchVAEConfig
-from .wavvae.model import WavVAE, WavVAEConfig
+# codec/models/__init__.py — keep inference-only imports
+from .patchvae.model import PatchVAE, PatchVAEConfig  # noqa: F401
+from .wavvae.model import WavVAE  # noqa: F401
+
+# IMPORTANT:
+# - Do NOT import pardi_tokenizer here (it references zcodec in your tree)
+# - Do NOT import training utilities here