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

andi4191 · andi4191 · commit 6eba01e05141 · 2022-03-31T15:03:14.000-07:00
Signed-off-by: Anurag Dixit &lt;a.dixit91@gmail.com&gt;
diff --git a/docker/WORKSPACE.docs b/docker/WORKSPACE.docs
@@ -45,6 +45,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)
 #############################################################################################################
