From c2f49547d802903b795986f7d5e6ee1cff985794 Mon Sep 17 00:00:00 2001
From: allegroai <>
Date: Mon, 4 Oct 2021 23:38:29 +0300
Subject: [PATCH] Fix broken Dataset finalize

---
 clearml/datasets/dataset.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/clearml/datasets/dataset.py b/clearml/datasets/dataset.py
index 12fc345d..a84f239f 100644
--- a/clearml/datasets/dataset.py
+++ b/clearml/datasets/dataset.py
@@ -1511,7 +1511,7 @@ class Dataset(object):
         #         self._dependency_graph.keys())
         #     return dict(chunks_lookup)
         chunks_lookup = map(
-            lambda d: (d, Dataset.get(dataset_id=d).get_num_chunks()),
+            lambda d: (d, (self if d == self.id else Dataset.get(dataset_id=d)).get_num_chunks(include_parents=False)),
             self._dependency_graph.keys())
         return dict(chunks_lookup)