point to proper import

wandb · Feb 7, 2024 · 5303186 · 5303186
1 parent 019ef73
commit 5303186
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/colabs/pytorch-lightning/Track_PyTorch_Lightning_with_Fabric_and_Wandb.ipynb b/colabs/pytorch-lightning/Track_PyTorch_Lightning_with_Fabric_and_Wandb.ipynb
@@ -72,8 +72,6 @@
       "outputs": [],
       "source": [
         "%%capture\n",
-        "# Remove when PR Merged\n",
-        "!pip install git+https://github.com/ash0ts/lightning.git@fabric-wandb\n",
         "!pip install wandb"
       ]
     },
@@ -115,7 +113,8 @@
       "source": [
         "import lightning as L\n",
         "import torch; import torchvision as tv\n",
-        "from lightning.fabric.loggers import WandbLogger"
+        "from wandb.integration.lightning.fabric import WandbLogger\n",
+        "\n"
       ]
     },
     {