diff --git a/DKT/trainer/__init__.py b/DKT/trainer/__init__.py
index 4662e75..f59450e 100644
--- a/DKT/trainer/__init__.py
+++ b/DKT/trainer/__init__.py
@@ -1,2 +1,3 @@
 from .trainer_ML import *
-from .trainer_GCN import *
\ No newline at end of file
+from .trainer_GCN import *
+from .trainer_HM import *
\ No newline at end of file