From 495a013451853381252b30abe9b80b751b1bb85a Mon Sep 17 00:00:00 2001
From: Yi Yang <yiya@google.com>
Date: Tue, 3 Sep 2024 03:27:59 -0700
Subject: [PATCH] Fix pip install command in tapnet pytorch colab.

PiperOrigin-RevId: 670484434
Change-Id: Ica0ebdc386368dd3ea94bc8425208f280df73d80
---
 colabs/torch_tapir_demo.ipynb | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/colabs/torch_tapir_demo.ipynb b/colabs/torch_tapir_demo.ipynb
index 35ecae6..6af12cc 100644
--- a/colabs/torch_tapir_demo.ipynb
+++ b/colabs/torch_tapir_demo.ipynb
@@ -66,7 +66,7 @@
       "outputs": [],
       "source": [
         "# @title Install code and dependencies {form-width: \"25%\"}\n",
-        "!pip install git+https://github.com/google-deepmind/tapnet.git[torch]"
+        "!pip install 'tapnet[torch] @ git+https://github.com/google-deepmind/tapnet.git'"
       ]
     },
     {