From eb123cfec9e81a6a88410049bc3836376ba0e68c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Fr=C3=A9d=C3=A9ric=20Branchaud-Charron?=
 <frederic.branchaud.charron@gmail.com>
Date: Thu, 27 Jun 2024 16:02:41 -0400
Subject: [PATCH] Update README.md

---
 README.md | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/README.md b/README.md
index e24ab10..584e80a 100644
--- a/README.md
+++ b/README.md
@@ -94,7 +94,7 @@ The framework consists of four main parts, as demonstrated in the flowchart belo
   <img src="docs/learn/literature/images/Baalscheme.svg">
 </p>
 
-To get started, wrap your dataset in our _[**ActiveLearningDataset**](baal/active/dataset.py)_ class. This will ensure
+To get started, wrap your dataset in our _[**ActiveLearningDataset**](baal/active/dataset/pytorch_dataset.py)_ class. This will ensure
 that the dataset is split into
 `training` and `pool` sets. The `pool` set represents the portion of the training set which is yet to be labelled.