truefoundry 0.4.8rc0__py3-none-any.whl → 0.4.9rc1__py3-none-any.whl

truefoundry/ml/artifact/truefoundry_artifact_repo.py

@@ -897,6 +897,9 @@ class MlFoundryArtifactsRepository:
             page_token = page.next_page_token
 
     def _is_directory(self, artifact_path):
+        # TODO: Ideally server should return a flag to indicate if it is a directory
+        # For now we can rely on list_artifacts because generally in cloud buckets
+        # directories are not stored as separate entities - empty directories ideally do not exist
         for _ in self.list_artifacts(artifact_path, page_size=3):
             return True
         return False
@@ -1014,7 +1017,9 @@ class MlFoundryArtifactsRepository:
 
     def _get_file_paths_recur(self, src_artifact_dir_path, dst_local_dir_path):
         local_dir = os.path.join(dst_local_dir_path, src_artifact_dir_path)
-        dir_content = [  # prevent infinite loop, sometimes the dir is recursively included
+        # prevent infinite loop, sometimes the dir is recursively included
+        # TODO (chiragjn): Check why and when this happens
+        dir_content = [
             file_info
             for file_info in self.list_artifacts(src_artifact_dir_path)
             if file_info.path != "." and file_info.path != src_artifact_dir_path
@@ -1099,8 +1104,8 @@ class MlFoundryArtifactsRepository:
 
         try:
             progress_bar.start()
-            # Check if the artifacts points to a directory
-            if self._is_directory(artifact_path):
+            # empty string is treated as root directory
+            if not artifact_path or self._is_directory(artifact_path):
                 futures: List[Future] = []
                 file_paths: List[Tuple[str, str]] = []
                 abort_event = Event()

{truefoundry-0.4.8rc0.dist-info → truefoundry-0.4.9rc1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: truefoundry
-Version: 0.4.8rc0
+Version: 0.4.9rc1
 Summary: Truefoundry CLI
 Author: Abhishek Choudhary
 Author-email: abhishek@truefoundry.com
@@ -37,7 +37,7 @@ Requires-Dist: python-dateutil (>=2.8.2,<3.0.0)
 Requires-Dist: python-dotenv (>=1.0.1,<2.0.0)
 Requires-Dist: python-socketio[client] (>=5.5.2,<6.0.0)
 Requires-Dist: questionary (>=1.10.0,<2.0.0)
-Requires-Dist: requests (>=2.31.0,<3.0.0)
+Requires-Dist: requests (>=2.18.0,<3.0.0)
 Requires-Dist: requirements-parser (>=0.11.0,<0.12.0)
 Requires-Dist: rich (>=13.7.1,<14.0.0)
 Requires-Dist: rich-click (>=1.2.1,<2.0.0)

{truefoundry-0.4.8rc0.dist-info → truefoundry-0.4.9rc1.dist-info}/RECORD

@@ -125,7 +125,7 @@ truefoundry/langchain/utils.py,sha256=PGLDe9chZ3BuUjakexOGpIqZRFoHEgu-zJ9yKdpLLm
 truefoundry/logger.py,sha256=u-YCNjg5HBwE70uQcpjIG64Ghos-K2ulTWaxC03BSj4,714
 truefoundry/ml/__init__.py,sha256=2A1l7pgqbVRt3cRW_0Lxg92hyJEkMxkCUh1EFprrmc0,942
 truefoundry/ml/artifact/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-truefoundry/ml/artifact/truefoundry_artifact_repo.py,sha256=VU8i3jnY62MLfzA3rxXuUjdqLz8Yaw4zqqPWSsf0mBg,45850
+truefoundry/ml/artifact/truefoundry_artifact_repo.py,sha256=FksxhUpRHb9pgWZmAB16DhXqkAL6UIAPA1S3RJUApQU,46201
 truefoundry/ml/autogen/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/ml/autogen/client/__init__.py,sha256=B6Z0oUmE-S9YqUCDRpKDuTyNWX2Fmx4es7nWx7Bhy10,17600
 truefoundry/ml/autogen/client/api/__init__.py,sha256=3sMSljMIS3UHYeF0BcNvrPPx6VbBSRt_1IfDn-13Kyc,752
@@ -355,7 +355,7 @@ truefoundry/workflow/remote_filesystem/tfy_signed_url_client.py,sha256=5mBCIc-ON
 truefoundry/workflow/remote_filesystem/tfy_signed_url_fs.py,sha256=Hf6Dk6Fu6P7DqsK5ULgraf9DStjgigf-kjaRAMBW-RU,8680
 truefoundry/workflow/task.py,sha256=ToitYiKcNzFCtOVQwz1W8sRjbR97eVS7vQBdbgUQtKg,1779
 truefoundry/workflow/workflow.py,sha256=WaTqUjhwfAXDWu4E5ehuwAxrCbDJkoAf1oWmR2E9Qy0,4575
-truefoundry-0.4.8rc0.dist-info/METADATA,sha256=29yqtR3hnoaIvy_2VfJprq87Jj1ycisFuMBTYVhGrTY,3102
-truefoundry-0.4.8rc0.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-truefoundry-0.4.8rc0.dist-info/entry_points.txt,sha256=TXvUxQkI6zmqJuycPsyxEIMr3oqfDjgrWj0m_9X12x4,95
-truefoundry-0.4.8rc0.dist-info/RECORD,,
+truefoundry-0.4.9rc1.dist-info/METADATA,sha256=9IU6anotmcm0swYYWFYpLojpa5xBZaGzLB59yczTOo0,3102
+truefoundry-0.4.9rc1.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+truefoundry-0.4.9rc1.dist-info/entry_points.txt,sha256=TXvUxQkI6zmqJuycPsyxEIMr3oqfDjgrWj0m_9X12x4,95
+truefoundry-0.4.9rc1.dist-info/RECORD,,