Skip to content

Commit

Permalink
Update tpv/core/helpers.py
Browse files Browse the repository at this point in the history
Co-authored-by: Nuwan Goonasekera <2070605+nuwang@users.noreply.github.com>
  • Loading branch information
sanjaysrikakulam and nuwang authored Jan 24, 2024
1 parent 3c44b41 commit 8aa024d
Showing 1 changed file with 1 addition and 5 deletions.
6 changes: 1 addition & 5 deletions tpv/core/helpers.py
Original file line number Diff line number Diff line change
Expand Up @@ -119,8 +119,4 @@ def tool_version_gt(tool, version):
def object_store_ids_and_dataset_size(job):
# Return a dictionary of dataset ids and their object store ids
# and file sizes in bytes for all input datasets in a job
object_store_ids = {}
for i in job.get_input_datasets():
object_store_ids[i.dataset.dataset.id] = (i.dataset.dataset.object_store_id, float(i.dataset.dataset.get_size(calculate_size=False)))

return object_store_ids
return {i.dataset.dataset.id: {'object_store_id': i.dataset.dataset.object_store_id, 'size': get_dataset_size(i.dataset.dataset)} for i in datasets or {}}

0 comments on commit 8aa024d

Please sign in to comment.