From cf083546aa19ece3791251d08d4216be9a100cca Mon Sep 17 00:00:00 2001
From: liu-yongqi-63 <liuyongqi5@huawei.com>
Date: Wed, 15 Feb 2023 11:30:49 +0800
Subject: [PATCH] Provide len() method for the minddata dataset class interface

---
 mindspore/python/mindspore/dataset/engine/datasets.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/mindspore/python/mindspore/dataset/engine/datasets.py b/mindspore/python/mindspore/dataset/engine/datasets.py
index 009d8e4cf45..aeba22967a6 100644
--- a/mindspore/python/mindspore/dataset/engine/datasets.py
+++ b/mindspore/python/mindspore/dataset/engine/datasets.py
@@ -1853,6 +1853,15 @@ class Dataset:
     def parse(self, children=None):
         raise NotImplementedError("Dataset has to implement parse method.")
 
+    def __len__(self):
+        """
+        Get the length of dataset.
+
+        Returns:
+            int, the length of dataset.
+        """
+        return self.get_dataset_size()
+
     @staticmethod
     def _update_data_shard(num_shards, shard_id):
         """