diff --git a/tensorflow_gnn/BUILD b/tensorflow_gnn/BUILD
index bc8b108e..fb07c875 100644
--- a/tensorflow_gnn/BUILD
+++ b/tensorflow_gnn/BUILD
@@ -46,3 +46,8 @@ pytype_strict_library(
     srcs_version = "PY3",
     visibility = ["//visibility:private"],
 )
+
+filegroup(
+    name = "files",
+    srcs = glob(["**"]),
+)