Update all files for SkySensepp

s2/sky_sensepp_impl/necks/__init__.py

@@ -1,12 +1,13 @@
 from .transformer_encoder import TransformerEncoder
 from .modality_completion import ModalityCompletion
-from .modality_vae import ModalityCompletionVAE
+from .modality_vae import LegacyModalityCompletionVAE, ModalityCompletionVAE
 from .modality_vae_config import ModalityCompletionVAEConfig
 from .modality_vae_pipeline import ModalityCompletionVAEPipeline
 
 __all__ = [
     "TransformerEncoder",
     "ModalityCompletion",
+    "LegacyModalityCompletionVAE",
     "ModalityCompletionVAE",
     "ModalityCompletionVAEConfig",
     "ModalityCompletionVAEPipeline",