make predict helper test pass in bazel

nutonomy · Mar 6, 2024 · a932818 · a932818
1 parent 68d8486
commit a932818
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/python-sdk/nuscenes/prediction/BUILD b/python-sdk/nuscenes/prediction/BUILD
@@ -0,0 +1,11 @@
+py_library(
+    name = "prediction",
+    srcs = [
+        "__init__.py",
+        "helper.py",
+    ],
+    deps = [
+        "//python-sdk/nuscenes/eval",
+    ],
+    visibility = ["//visibility:public"],
+)
diff --git a/python-sdk/nuscenes/tests/BUILD b/python-sdk/nuscenes/tests/BUILD
@@ -40,7 +40,7 @@ py_test(
     visibility = ["//visibility:public"],
     deps = [
         "//python-sdk/nuscenes",
-        # TODO add nuscenes.prediction
+        "//python-sdk/nuscenes/prediction",
     ],
     imports = ["../.."], # add this to PYTHONPATH, relative to this BUILD file
     env = {