From 1c84b310564583740d123223fc83d758174c4835 Mon Sep 17 00:00:00 2001
From: allegroai <>
Date: Sun, 24 Jan 2021 09:21:28 +0200
Subject: [PATCH] Fix Dataset generate removed / modified / added of summary
 table

---
 clearml/datasets/dataset.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/clearml/datasets/dataset.py b/clearml/datasets/dataset.py
index 556a6289..b99c5e2a 100644
--- a/clearml/datasets/dataset.py
+++ b/clearml/datasets/dataset.py
@@ -504,7 +504,7 @@ class Dataset(object):
         :return: List of files with relative path
             (files might not be available locally until get_local_copy() is called)
         """
-        datasets = self._dependency_graph[self._id] if not dataset_id else [dataset_id]
+        datasets = self._dependency_graph[self._id] if not dataset_id or dataset_id == self._id else [dataset_id]
         unified_list = set()
         for ds_id in datasets:
             dataset = self.get(dataset_id=ds_id)
@@ -522,7 +522,7 @@ class Dataset(object):
         :return: List of files with relative path
             (files might not be available locally until get_local_copy() is called)
         """
-        datasets = self._dependency_graph[self._id] if not dataset_id else [dataset_id]
+        datasets = self._dependency_graph[self._id] if not dataset_id or dataset_id == self._id else [dataset_id]
         unified_list = dict()
         for ds_id in datasets:
             dataset = self.get(dataset_id=ds_id)
@@ -541,7 +541,7 @@ class Dataset(object):
         :return: List of files with relative path
             (files might not be available locally until get_local_copy() is called)
         """
-        datasets = self._dependency_graph[self._id] if not dataset_id else [dataset_id]
+        datasets = self._dependency_graph[self._id] if not dataset_id or dataset_id == self._id else [dataset_id]
         unified_list = set()
         for ds_id in datasets:
             dataset = self.get(dataset_id=ds_id)