From 435331f0b2912fc66b93878efd10d75ba93fd52a Mon Sep 17 00:00:00 2001
From: SachaSamama <sachasamama@gmail.com>
Date: Thu, 21 Oct 2021 11:13:38 +0200
Subject: [PATCH] Update utils.py (#233)

---
 darwin/torch/utils.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/darwin/torch/utils.py b/darwin/torch/utils.py
index a02c1512b..20b5f317c 100644
--- a/darwin/torch/utils.py
+++ b/darwin/torch/utils.py
@@ -100,7 +100,9 @@ def detectron2_register_dataset(
     catalog_name = f"darwin_{dataset_path.name}"
     if partition:
         catalog_name += f"_{partition}"
-    classes = get_classes(dataset_path, annotation_type="polygon")
+        
+    classes = get_classes(dataset_path=dataset_path, release_name=release_name, annotation_type="polygon")
+    
     DatasetCatalog.register(
         catalog_name,
         lambda partition=partition: list(