(//docker): Fixing WORKSPACE.docs for CI

andi4191 · andi4191 · commit dd99bd47e328 · 2022-04-18T14:53:46.000-07:00
Signed-off-by: Anurag Dixit &lt;a.dixit91@gmail.com&gt;
diff --git a/docker/WORKSPACE.docs b/docker/WORKSPACE.docs
@@ -44,6 +44,11 @@ new_local_repository(
     build_file = "@//third_party/cublas:BUILD",
 )
 
+local_repository(
+    name = "torch_tensorrt",
+    path = "/usr/local/lib/python3.8/dist-packages/torch_tensorrt"
+)
+
 #############################################################################################################
 # Tarballs and fetched dependencies (default - use in cases when building from precompiled bin and tarballs)
 #############################################################################################################
