diff --git a/emote/memory/__init__.py b/emote/memory/__init__.py index ba3109a2..1d9f2a78 100644 --- a/emote/memory/__init__.py +++ b/emote/memory/__init__.py @@ -4,6 +4,7 @@ from .callbacks import MemoryImporterCallback from .memory import ( + JointMemoryLoader, LoggingProxyWrapper, MemoryExporterProxyWrapper, MemoryLoader, @@ -21,4 +22,5 @@ "MemoryImporterCallback", "LoggingProxyWrapper", "MemoryWarmup", + "JointMemoryLoader", ]