diff --git a/requirements-swh.txt b/requirements-swh.txt
--- a/requirements-swh.txt
+++ b/requirements-swh.txt
@@ -1,2 +1,3 @@
 swh.core[http] >= 0.3
 swh.model >= 0.13.0
+swh.dataset
diff --git a/swh/graph/tests/dataset/generate_dataset.py b/swh/graph/tests/dataset/generate_dataset.py
--- a/swh/graph/tests/dataset/generate_dataset.py
+++ b/swh/graph/tests/dataset/generate_dataset.py
@@ -1,5 +1,7 @@
 #!/usr/bin/env python3
 
+# type: ignore
+
 # Copyright (C) 2021  The Software Heritage developers
 # See the AUTHORS file at the top-level directory of this distribution
 # License: GNU General Public License version 3, or any later version