python/pytorch: Fix regression from 6904

Signed-off-by: Soham Manoli <[email protected]>
NVIDIA · Jun 7, 2024 · 5429a7f · 5429a7f
1 parent 63bfb42
commit 5429a7f
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/python/aistore/pytorch/utils.py b/python/aistore/pytorch/utils.py
@@ -34,7 +34,7 @@ def list_objects(
     for url in urls_list:
         provider, bck_name, path = parse_url(url)
         bucket = client.bucket(bck_name=bck_name, provider=provider)
-        samples.extend([obj.name for obj in bucket.list_all_objects_iter(prefix=path)])
+        samples.extend([obj for obj in bucket.list_all_objects_iter(prefix=path)])
 
     for source in ais_source_list:
         samples.extend([obj.name for obj in source.list_all_objects_iter()])

diff --git a/python/tests/integration/pytorch/test_pytorch_plugin.py b/python/tests/integration/pytorch/test_pytorch_plugin.py
@@ -175,7 +175,7 @@ def test_multishard_stream(self):
             bucket_name=self.bck_name,
             prefix="test_multishard_shard2.tar",
         )
-        dataset = AISMultiShardStream(data_sorces=[shard1, shard2])
+        dataset = AISMultiShardStream(data_sources=[shard1, shard2])
         combined_content = list(
             zip(shard1_content_dict.values(), shard2_content_dict.values())
         )

diff --git a/python/tests/unit/pytorch/test_datasets.py b/python/tests/unit/pytorch/test_datasets.py
@@ -68,7 +68,7 @@ def test_multi_shard_stream(self):
 
         self.shards = [MagicMock(), MagicMock(), MagicMock()]
 
-        stream = AISMultiShardStream(data_sorces=self.shards)
+        stream = AISMultiShardStream(data_sources=self.shards)
 
         expected_results = [
             (b"data1_1", b"data2_1", b"data3_1"),