Fixing model file listing issue

harsh97 · harsh97 · commit 94d3501f04ea · 2024-07-15T11:13:10.000+05:30
diff --git a/ads/aqua/common/utils.py b/ads/aqua/common/utils.py
@@ -251,9 +251,9 @@ def list_os_files_with_extension(oss_path: str, extension: str) -> [str]:
     files: List[ObjectSummary] = oss_client.list_objects().objects
 
     return [
-        file.name
+        file.name[len(oss_client.filepath) :]
         for file in files
-        if file.name.endswith(extension) and "/" not in file.name
+        if file.name.endswith(extension)
     ]
 
 

Original file line number	Diff line number	Diff line change
`@@ -251,9 +251,9 @@ def list_os_files_with_extension(oss_path: str, extension: str) -> [str]:`
`251`	`251`	`files: List[ObjectSummary] = oss_client.list_objects().objects`
`252`	`252`
`253`	`253`	`return [`
`254`		`- file.name`
	`254`	`+ file.name[len(oss_client.filepath) :]`
`255`	`255`	`for file in files`
`256`		`- if file.name.endswith(extension) and "/" not in file.name`
	`256`	`+ if file.name.endswith(extension)`
`257`	`257`	`]`
`258`	`258`
`259`	`259`