To run it on Google Colab change the first code block to this:
!git clone https://github.com/dubajstan/Prototypical-Networks.git
%cd Prototypical-Networks
!pip install uv
!uv pip install -e .
import sys
sys.path.append("/content/Prototypical-Networks")
sys.path.append("/content/Prototypical-Networks/data")
sys.path.append("/content/Prototypical-Networks/src")
sys.path.append("/content/Prototypical-Networks/configs")
sys.path.append("/content/Prototypical-Networks/notebooks")
import torch
from pathlib import Path
import sys
device = "cuda" if torch.cuda.is_available() else "cpu"
print(f"Using device: {device}")
BASE_PATH = Path.cwd() # deleted .parent
DATA_PATH = BASE_PATH / "data"
sys.path.append(str(BASE_PATH))