diff --git a/setup.py b/setup.py
index 224e13980d66eb209b86dbb0fe5847da064f42a8..128fb589cd7929247f709c9931184aa647a73a82 100644
--- a/setup.py
+++ b/setup.py
@@ -78,6 +78,7 @@ setup(
         'train_conditionalgan_casia.py = bob.learn.pytorch.scripts.train_conditionalgan_casia:main', 
         'train_wcgan_multipie.py = bob.learn.pytorch.scripts.train_wcgan_multipie:main', 
         'train_drgan_multipie.py = bob.learn.pytorch.scripts.train_drgan_multipie:main', 
+        'read_training_hdf5.py = bob.learn.pytorch.scripts.read_training_hdf5:main', 
       ],