from pilot.source_embedding.source_embedding import SourceEmbedding, register __all__ = ["SourceEmbedding", "register"]