Browse Source

rename 'input' to 'inputs' in mocked Dataset

tags/v1.1.0
luopengting 5 years ago
parent
commit
4aa89d5fe4
2 changed files with 2 additions and 2 deletions
  1. +1
    -1
      tests/utils/lineage_writer/model_lineage.py
  2. +1
    -1
      tests/utils/mindspore/dataset/engine/datasets.py

+ 1
- 1
tests/utils/lineage_writer/model_lineage.py View File

@@ -529,7 +529,7 @@ class AnalyzeObject:
return output_dataset.dataset_dir
if isinstance(output_dataset, dataset_files_set):
return output_dataset.dataset_files[0]
return self.get_dataset_path(output_dataset.input[0])
return self.get_dataset_path(output_dataset.inputs[0])

@staticmethod
def get_dataset_path_wrapped(dataset):


+ 1
- 1
tests/utils/mindspore/dataset/engine/datasets.py View File

@@ -21,7 +21,7 @@ class Dataset:
def __init__(self, dataset_size=None, dataset_path=None):
self.dataset_size = dataset_size
self.dataset_path = dataset_path
self.input = []
self.inputs = []

def get_dataset_size(self):
"""Mocked get_dataset_size."""


Loading…
Cancel
Save