trainml 0.5.17__py3-none-any.whl → 1.0.1__py3-none-any.whl

trainml/volumes.py

@@ -10,7 +10,7 @@ from .exceptions import (
     SpecificationError,
     TrainMLException,
 )
-from .connections import Connection
+from trainml.utils.transfer import upload, download
 
 
 class Volumes(object):
@@ -56,7 +56,9 @@ class Volumes(object):
         return volume
 
     async def remove(self, id, **kwargs):
-        await self.trainml._query(f"/volume/{id}", "DELETE", dict(**kwargs, force=True))
+        await self.trainml._query(
+            f"/volume/{id}", "DELETE", dict(**kwargs, force=True)
+        )
 
 
 class Volume:
@@ -120,56 +122,45 @@ class Volume:
         )
         return resp
 
-    async def get_connection_utility_url(self):
-        resp = await self.trainml._query(
-            f"/volume/{self._id}/download",
-            "GET",
-            dict(project_uuid=self._project_uuid),
-        )
-        return resp
-
-    def get_connection_details(self):
-        if self._volume.get("vpn"):
-            details = dict(
-                entity_type="volume",
-                project_uuid=self._volume.get("project_uuid"),
-                cidr=self._volume.get("vpn").get("cidr"),
-                ssh_port=self._volume.get("vpn").get("client").get("ssh_port"),
-                input_path=(
-                    self._volume.get("source_uri")
-                    if self.status in ["new", "downloading"]
-                    else None
-                ),
-                output_path=(
-                    self._volume.get("output_uri")
-                    if self.status == "exporting"
-                    else None
-                ),
-            )
-        else:
-            details = dict()
-        return details
-
     async def connect(self):
-        if self.status in ["ready", "failed"]:
-            raise SpecificationError(
-                "status",
-                f"You can only connect to downloading or exporting volumes.",
-            )
-        if self.status == "new":
-            await self.wait_for("downloading")
-        connection = Connection(
-            self.trainml, entity_type="volume", id=self.id, entity=self
-        )
-        await connection.start()
-        return connection.status
+        if self.status not in ["downloading", "exporting"]:
+            if self.status == "new":
+                await self.wait_for("downloading")
+            else:
+                raise SpecificationError(
+                    "status",
+                    f"You can only connect to downloading or exporting volumes.",
+                )
 
-    async def disconnect(self):
-        connection = Connection(
-            self.trainml, entity_type="volume", id=self.id, entity=self
-        )
-        await connection.stop()
-        return connection.status
+        # Refresh to get latest entity data
+        await self.refresh()
+
+        if self.status == "downloading":
+            # Upload task - get auth_token, hostname, and source_uri from volume
+            auth_token = self._volume.get("auth_token")
+            hostname = self._volume.get("hostname")
+            source_uri = self._volume.get("source_uri")
+
+            if not auth_token or not hostname or not source_uri:
+                raise SpecificationError(
+                    "status",
+                    f"Volume in downloading status missing required connection properties (auth_token, hostname, source_uri).",
+                )
+
+            await upload(hostname, auth_token, source_uri)
+        elif self.status == "exporting":
+            # Download task - get auth_token, hostname, and output_uri from volume
+            auth_token = self._volume.get("auth_token")
+            hostname = self._volume.get("hostname")
+            output_uri = self._volume.get("output_uri")
+
+            if not auth_token or not hostname or not output_uri:
+                raise SpecificationError(
+                    "status",
+                    f"Volume in exporting status missing required connection properties (auth_token, hostname, output_uri).",
+                )
+
+            await download(hostname, auth_token, output_uri)
 
     async def remove(self, force=False):
         await self.trainml._query(
@@ -208,7 +199,9 @@ class Volume:
                 if msg_handler:
                     msg_handler(data)
                 else:
-                    timestamp = datetime.fromtimestamp(int(data.get("time")) / 1000)
+                    timestamp = datetime.fromtimestamp(
+                        int(data.get("time")) / 1000
+                    )
                     print(
                         f"{timestamp.strftime('%m/%d/%Y, %H:%M:%S')}: {data.get('msg').rstrip()}"
                     )
@@ -237,7 +230,7 @@ class Volume:
     async def wait_for(self, status, timeout=300):
         if self.status == status:
             return
-        valid_statuses = ["downloading", "ready", "archived"]
+        valid_statuses = ["downloading", "ready", "exporting", "archived"]
         if not status in valid_statuses:
             raise SpecificationError(
                 "status",
@@ -252,7 +245,9 @@ class Volume:
             )
         POLL_INTERVAL_MIN = 5
         POLL_INTERVAL_MAX = 60
-        POLL_INTERVAL = max(min(timeout / 60, POLL_INTERVAL_MAX), POLL_INTERVAL_MIN)
+        POLL_INTERVAL = max(
+            min(timeout / 60, POLL_INTERVAL_MAX), POLL_INTERVAL_MIN
+        )
         retry_count = math.ceil(timeout / POLL_INTERVAL)
         count = 0
         while count < retry_count:

{trainml-0.5.17.dist-info → trainml-1.0.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: trainml
-Version: 0.5.17
+Version: 1.0.1
 Summary: trainML client SDK and command line utilities
 Home-page: https://github.com/trainML/trainml-cli
 Author: trainML
@@ -12,8 +12,8 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.8
 Requires-Python: >=3.8
 Description-Content-Type: text/markdown
-Requires-Dist: aiodocker
 Requires-Dist: aiohttp
+Requires-Dist: aiofiles
 Requires-Dist: boto3
 Requires-Dist: Click (>8)
 Requires-Dist: python-jose[cryptography]
@@ -190,7 +190,7 @@ To list all datasets:
 trainml dataset list
 ```
 
-To connect to a job that requires the [connection capability](https://docs.trainml.ai/reference/connection-capability):
+To connect to a job that uses the "local" file transfer method:
 
 ```
 trainml job connect <job ID or name>

{trainml-0.5.17.dist-info → trainml-1.0.1.dist-info}/RECORD

@@ -1,16 +1,16 @@
 examples/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 examples/create_dataset_and_training_job.py,sha256=Ht6soyBZxt8qA9Fp6zAW2syGEdasvyYp-qvQqE18g6k,1178
-examples/local_storage.py,sha256=w8iAeqr5CLOCOkNrqGzEDtybjDGGY7SQUqeE0ibMUrM,1745
+examples/local_storage.py,sha256=vI67RbDjWvXovUv9ahoj81KJIJxXdIU2J4PwT_vCalI,1687
 examples/training_inference_pipeline.py,sha256=SNr4RFT9y69F9G9tMD8ONUbJmXRFrq1yxynq-FbfEf8,2334
 tests/integration/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tests/integration/conftest.py,sha256=otgb7lZu62_5zAMAsPpr0U2j0h5lApPlDRCiu7ZfDr8,1135
-tests/integration/test_checkpoints_integration.py,sha256=RbOHZK-fjj2xvYo1XDa6H0war43Qe_1PbiNFndVY87o,3239
-tests/integration/test_datasets_integration.py,sha256=sV88UAMM0IyVVum3zvxfYFbqmR8W4skNWE7jQxbc9ms,3529
+tests/integration/test_checkpoints_integration.py,sha256=pUu8PyBdnpJjHhcFe76SqgOvhNbNAiLk1dFXvKiwPpg,3219
+tests/integration/test_datasets_integration.py,sha256=TfyAMBjLrIVRW-3h0mox69ll7P-_5KQMSyX8toTR2k0,3518
 tests/integration/test_environments_integration.py,sha256=0IckhJvQhd8j4Ouiu0hMq2b7iA1dbZpZYmknyfWjsFM,1403
 tests/integration/test_gpu_types_integration.py,sha256=V2OncokZWWVq_l5FSmKEDM4EsWrmpB-zKiVPt-we0aY,1256
-tests/integration/test_jobs_integration.py,sha256=_c33hYpub3VoWiFY-dAzTVyB66U3n5yvMxoRR-pskHg,25263
-tests/integration/test_models_integration.py,sha256=Kwh8MQ0hOaecclAPq07LBEtAxoZB3rx-yamq0WGtWEA,2906
-tests/integration/test_volumes_integration.py,sha256=W5DeuuKCBx3C89LRWc0jhAZjJEREuPiawTLUnqpPrjg,3299
+tests/integration/test_jobs_integration.py,sha256=AQnnoSR75Qqm2XouEUCiVt-6wQbxf8c7iVY9c16lAmQ,25386
+tests/integration/test_models_integration.py,sha256=VAyiXel3E8dAizDALpYYtgazek07y-tIErIhGIS8BXc,2924
+tests/integration/test_volumes_integration.py,sha256=0_D0-mY72_UdZH6clwl-hGwg-yqF8c7mJ70cyyHaBH0,3268
 tests/integration/cloudbender/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tests/integration/cloudbender/conftest.py,sha256=jBYUHKJioI-WHM-d0Y4lq5pD4LiOPINj6K9kQkgaGWY,777
 tests/integration/cloudbender/test_providers_integration.py,sha256=mQn2dr4oMpctCVCVs53G5P0Raa0U2uBxEYjZtFFSvTg,1327
@@ -26,21 +26,22 @@ tests/integration/projects/test_projects_members_integration.py,sha256=bEc9AqC-Y
 tests/integration/projects/test_projects_secrets_integration.py,sha256=dnIT0KiHnXBVP-azfw8MCspTjX7q1ic03rz8XZNijvU,1520
 tests/integration/projects/test_projects_services_integration.py,sha256=3DpOtVMIdG_Icp4s7JxMT70ZDKC1wMD8imkKjlQChb0,1569
 tests/unit/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-tests/unit/conftest.py,sha256=hLfECgT0w4pVZABrpK4UUUeaX1aDzQvj8-erWTGBjHA,35588
-tests/unit/test_auth_unit.py,sha256=nfhlOCR7rUsn_MaD8QQtBc2v0k8pIxqbzGgRAZK1WGc,858
-tests/unit/test_checkpoints_unit.py,sha256=7bgpEnjkfETgbEHmYApyD_htEWAUVZdposyYVClU3tQ,15965
+tests/unit/conftest.py,sha256=7pLqYwSo2ApqWNtBTj7SQwSGl1R2haxOCoUyUzoFkl8,35592
+tests/unit/test_auth_unit.py,sha256=o6mqzEmqWqvWirMitcHZtn_ScNoQ5jZ20HWrA4OVnPs,1413
+tests/unit/test_checkpoints_unit.py,sha256=dAgzXVmdqUyYrzysWM8GrlEM4xJYOf-IH7WB4v87IC4,21914
 tests/unit/test_connections_unit.py,sha256=FzN2ddQxNpjxzNGUsXhjTk0HnD24wSPelPTL4o_r-Ho,5507
-tests/unit/test_datasets_unit.py,sha256=YL_wU-NiXvPFOK_jM8l9q5cZGztsMkgQorggYaLt6Lc,15910
+tests/unit/test_datasets_unit.py,sha256=xyaHdqDzHetGA2QQ1Vh4j-h_FjC-EC5pb3VTay8sfGY,21135
 tests/unit/test_environments_unit.py,sha256=1QFGf1xwM0yKCyVHT_Xi0DX8g0Aelr0mcqAImEXJfQU,1882
-tests/unit/test_exceptions.py,sha256=3tAok6kAU1QRjN7qTNVYuSGWDg7IEoK__OXFLyzLr7k,906
-tests/unit/test_gpu_types_unit.py,sha256=c9ie6YSYT5onBnlmHvHWON9WgQiJ1eO2C-4Tk-UPQHg,2054
-tests/unit/test_jobs_unit.py,sha256=bZxN9HUfHCyQCjZCZGn6WFIhu8S5FU1z5ZG9sgH2XEg,26835
-tests/unit/test_models_unit.py,sha256=Wg2QsoKqtSFSVf-1tt4sA6OcwMTkSHYPcUBQQOmZCgY,15091
-tests/unit/test_trainml_unit.py,sha256=8vAKvFD1xYsx_VY4HFVa0b1MUlMoNApY6TO8r7vI-UQ,1701
-tests/unit/test_volumes_unit.py,sha256=DF3kFtkIfuZjwacBRpdmMGlPvV7vxaxUAhlstWj2V4w,15382
+tests/unit/test_exceptions.py,sha256=uAhCXjbzA_m4No49bzbbBoYtvV0TldvxYYJPikcMZH0,5749
+tests/unit/test_gpu_types_unit.py,sha256=KmlmckRb-Faw8vfCl95HAHB51qei9HNElcN16V2gJfg,2526
+tests/unit/test_jobs_unit.py,sha256=byjdqe1TGEPCp5ExU5IS1tcu9eGotf1s2j-lA3FO058,63964
+tests/unit/test_main_unit.py,sha256=aC4-BsdE_bQoR61c0z0ZWOgfCaphSjgVyW1hIZqbSfk,627
+tests/unit/test_models_unit.py,sha256=S-WwRH-I3XE_ie6u-GQtwo95v-X6YeEBl5QFB76dauE,20127
+tests/unit/test_trainml_unit.py,sha256=N8yMUiSzA5A6ewMH-R2y9O5U8LDQwzEnHMLlwl4DjXI,23756
+tests/unit/test_volumes_unit.py,sha256=MH7CddtRFhH3Bn3k_SPVyt0EiramjLAYP0Kn4n4Sz5E,20185
 tests/unit/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tests/unit/cli/conftest.py,sha256=w6p_2URicywJKUCtY79tSD_mx8cwJtxHbK_Lu3grOYs,236
-tests/unit/cli/test_cli_checkpoint_unit.py,sha256=6gO6PWWxNJiL520mhTFkSUJT7p5-dptLwUgDjPeHNrA,702
+tests/unit/cli/test_cli_checkpoint_unit.py,sha256=2MGaSRmjechSgcgwQUQ1HwasROMnf93icLJRpjUOWek,11830
 tests/unit/cli/test_cli_datasets_unit.py,sha256=46LSaGHn21r3Xsd8GUDMTXTqU83m4puLY7U5XehkbMk,653
 tests/unit/cli/test_cli_environment_unit.py,sha256=7FFKFPVa6yqJqujTXD_tsSUDruW5KhlLgShPZTn1Eio,683
 tests/unit/cli/test_cli_gpu_unit.py,sha256=FIq3tQIDmeD-pvxkhJKMykfnlWcVxho2vRkoCMXgQxE,650
@@ -66,39 +67,41 @@ tests/unit/cloudbender/test_data_connectors_unit.py,sha256=qy97mNAcy_xEkh8obBobH
 tests/unit/cloudbender/test_datastores_unit.py,sha256=5XUczUrOQ2Zgfb2IbJeEtYxbBWRltEVJ1KIC_UeQjEY,5281
 tests/unit/cloudbender/test_device_configs_unit.py,sha256=lzyCuF7MRoQrtJVTQFL27lqPnRwQFv25htPgJqDuQI8,5714
 tests/unit/cloudbender/test_devices_unit.py,sha256=QBWnlOe1tw_XNA_i-yDHkmpGvtK36f2u1HhoXquoVaE,9103
-tests/unit/cloudbender/test_nodes_unit.py,sha256=BDpfJXCBNNpLt5rhJMk2BVXDQ_4QSmxoVrO_YPs6xBU,6231
-tests/unit/cloudbender/test_providers_unit.py,sha256=OgxifgC1IqLH8DNMKXy1Ne9_7a75ea6kHEOfRSRoQuQ,4373
-tests/unit/cloudbender/test_regions_unit.py,sha256=BbJICLIQmlotpA1UmLD0KTW_H9g2UW0J8ZYzQk1_Xjc,6299
-tests/unit/cloudbender/test_services_unit.py,sha256=fYJx-W89HD-EYkO32_v33X40VxipUfWQCy13FZO2fcA,5220
+tests/unit/cloudbender/test_nodes_unit.py,sha256=xiXHixKFH0NL1b4sp4oT_0XiLIDifCsglVloZSaWXdc,10928
+tests/unit/cloudbender/test_providers_unit.py,sha256=lDKYGP2pROxb-dtDVPz5Sz55ggnK68hAvsIy6AzL5tw,8588
+tests/unit/cloudbender/test_regions_unit.py,sha256=DJhkBJ-5jHKxCwCAdtV041xzIfjTGrKE77DzEFX-5Nc,10961
+tests/unit/cloudbender/test_services_unit.py,sha256=R4Vu3lq2jT_qQaxwRk449oqAhYrMFbCNp3KS4Pke_Hc,10956
 tests/unit/projects/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tests/unit/projects/test_project_credentials_unit.py,sha256=ObTaJ7_g0MP5N-0qHWdeYips0in1j_4UkOhXXI2FFwo,3480
-tests/unit/projects/test_project_data_connectors_unit.py,sha256=uQPFx6rwYMWZz1IJkns_LdM5gccyHSIaKb3k8zmfSkY,3385
-tests/unit/projects/test_project_datastores_unit.py,sha256=kcoaSs54gGqJ9gwPMpP6KHAPRvDJtdJpm49s4aYYt3E,3129
-tests/unit/projects/test_project_members_unit.py,sha256=C5N6jzT7zM3eL3_bXEddddBHPFJ3617T0lYluBjMy78,3466
+tests/unit/projects/test_project_data_connectors_unit.py,sha256=BceR9nqcrR_8fHXG2la4h4tMRrn2vUHptFDVhHCx0uU,5004
+tests/unit/projects/test_project_datastores_unit.py,sha256=VRl8ksYJKGa0m-PbBu1YVeBx_lIEry6Cu8mWPivQ_hs,4646
+tests/unit/projects/test_project_members_unit.py,sha256=OJbiRdDzUbhefiM5y5qDhjHzy0eRCwsWuwfV_2NEZsI,5097
 tests/unit/projects/test_project_secrets_unit.py,sha256=StVlY-ZR3CKxXAt9NL8hTkXUEdgGESH0m6foEPXKE-4,3276
-tests/unit/projects/test_project_services_unit.py,sha256=ZK5nz78RAmmaCrAwYBd34t87dh6etU-7snLB_xukZHM,3331
-tests/unit/projects/test_projects_unit.py,sha256=S4ydA9JLgGzFwaY4b30YJF0Pl6Iog4ZdzQmHpLy_UEA,3786
-trainml/__init__.py,sha256=an9i7gL01ZrVyaS0RF03wAeRJuWpgA1MgE8-tcwrH1c,433
+tests/unit/projects/test_project_services_unit.py,sha256=qGYVcs8CPF7DklPUQS3th8rZ42LGfeXiFsV__WTzUYg,6140
+tests/unit/projects/test_projects_unit.py,sha256=baI9-HRRSs4IPd82329zHWRGPYjNlJQ1qG4Hngcu6ds,4409
+tests/unit/utils/__init__.py,sha256=VDy3m4Lfazpey8NMzHWdgcs3T9aaXWuH7pglIjCnZW0,38
+tests/unit/utils/test_transfer_unit.py,sha256=seYUQwQK93kiZjfXEF_3ZN40za3xUTU61oXIyC4na0o,188309
+trainml/__init__.py,sha256=Jjeuhm0-YUNAgrdmG7fXrw6VgW19jN5yW0cz3JWwO98,432
 trainml/__main__.py,sha256=JgErYkiskih8Y6oRwowALtR-rwQhAAdqOYWjQraRIPI,59
 trainml/auth.py,sha256=clbx5S5prJ3u62aEESdBXIHF_HmreQ-L1ShxcyWQNDs,26565
-trainml/checkpoints.py,sha256=Hg3_Si7ohzsv4_8JIjoj9pMRpHNZRWKC5R2gtgmp01s,8790
+trainml/checkpoints.py,sha256=KEDS3xRZa1h5tsT6ZEsS5aanOfC-S_8yaeX52JubO4E,8924
 trainml/connections.py,sha256=h-S1NZbOkaXpIlpRStA6q-3eXc_OMlFWOLzF8R9SVG8,20029
-trainml/datasets.py,sha256=dxTZbK-cnr8eoq48KAx2TZNK1ndcFveTVBtaNK2QyJ0,8600
+trainml/datasets.py,sha256=T3z5iOMfFXYea002iPuVBURl0_Eia5nKLC4bOYmySXI,8690
 trainml/environments.py,sha256=OH4o08zXZ7IJ2CiA1rPnys2Fl45r8qvQHfM2mCBRAIc,1507
 trainml/exceptions.py,sha256=Qdof2fKRvbMiwarX1VSw1XJXXJjY71H4U3v05nE7-7g,5468
 trainml/gpu_types.py,sha256=mm-dwfYc02192bmYPIJmzesndyBcoOdkKYBaYZXOUwU,1901
-trainml/jobs.py,sha256=6iaZUz69v6NCSuM0jiSd6j8gElB3JOTuOZuVt2VtAHQ,17947
-trainml/models.py,sha256=SpVNt9oZMu70m_3VjO9j_QYexw_z-yRqMhIuJXPs4Vg,8284
-trainml/trainml.py,sha256=c-Qxiarv170RaQn0O1wJ6h9RP8e17M15KFg3Rh2ckZc,11902
-trainml/volumes.py,sha256=7T0ZN_Xq3qk4yzw7YYxC0rxH3KqlqADd6HOZCkAfCR4,8443
-trainml/cli/__init__.py,sha256=Gvj6oGSEtgpb40ACtiVeMD93GM-uy15MG6VlX6rwdwA,4346
-trainml/cli/checkpoint.py,sha256=8Rh4bmFwJ4DKlIjHK-FLTeRynABqKCgIUGRtbQhAsX4,7170
+trainml/jobs.py,sha256=1XzZrAQ13a35VDxG-x2gs2OKkgHdha_LbVrp-dlBQDE,26737
+trainml/models.py,sha256=PmbTNQKFlPqfRbq93UMpmo5za9qbNFmQdx6BKdejLXA,8339
+trainml/trainml.py,sha256=ZQy87waJniunebTSYfrOAlzIDY4bGxw8InPj7BjowtU,13688
+trainml/volumes.py,sha256=4Y8wWMymGWKOopRU3yV3NMyTrO2FrDJzdH7dN3hg8yQ,8533
+trainml/cli/__init__.py,sha256=rNiCPZpAeQSbZwceyQv0zmejzILVz5Km2lCiLw4uqjE,4330
+trainml/cli/checkpoint.py,sha256=feiqjeBn4HedBEQgPfENL8AtDeP76t2n-Sdcx9-A30o,6020
 trainml/cli/connection.py,sha256=ELV6bPL30dzttFNxDU7Fb74R8oPL_E70k7TcJEzbwtQ,1700
-trainml/cli/dataset.py,sha256=Pc00M6t7hGoRzCxznmmkijsWhG4PhIfG7UkrwtwykTY,6871
+trainml/cli/dataset.py,sha256=VNKjzYasKTUtVnOI58LW1Gq1VUyZcp7fTeQT9gw-1Xs,5721
 trainml/cli/environment.py,sha256=dfm_T8OlCM5M8dLyOQBapJl3eFuVIku2P4JO6V0cYVQ,1019
 trainml/cli/gpu.py,sha256=CMcQyl2qbUgc2bc-gvUVT6X7bq2-sgiCHl3hyZ6kFWM,883
-trainml/cli/model.py,sha256=hR23E6ttRXcLk-RofkPK6wUXMO7OU6sT6jTEHTmUg9Q,6111
-trainml/cli/volume.py,sha256=kDUss93N78DT-YlLjC6I3jEq5nBWfRNNR5M4tY_F_Zg,6246
+trainml/cli/model.py,sha256=Vb889aUxbwBXaA7ptRlIVMEBD8JaKLdE7qAzl0WZ8hE,4963
+trainml/cli/volume.py,sha256=T0hyBftupyES06YvZcG6LsqMvik8g7VkvotoDLwtAiQ,5135
 trainml/cli/cloudbender/__init__.py,sha256=tKkL8TzD9nEeRtf1OEYM4XZJWb0-rGMPTmLIdA5G_SY,592
 trainml/cli/cloudbender/data_connector.py,sha256=q0Hqeh5w00Zkmh61fzO3pNR6EmQfoLQ2DbgJ5ZBx1UM,3606
 trainml/cli/cloudbender/datastore.py,sha256=j67VUoYZOFWP0F7WK8V7fft8HlT4uqauAu0WH-9Dkes,3323
@@ -107,8 +110,8 @@ trainml/cli/cloudbender/node.py,sha256=iN_WaPCxOhtgDtnSsIFAEMGADG4MKiLjWoez6YSYw
 trainml/cli/cloudbender/provider.py,sha256=oFjZWKfFQjNY7OtDu7nUdfv-RTmQc_Huuug963D3BdA,1726
 trainml/cli/cloudbender/region.py,sha256=X6-FYOb-pGpOEazn-NbsYSwa9ergB7FGATFkTe4a8Pk,2892
 trainml/cli/cloudbender/service.py,sha256=Wh6ycEuECiKL7qpFhc4IyO1rR5lvLtIHk3S475_R6pk,3147
-trainml/cli/job/__init__.py,sha256=ljY-ELeXhXQ7txASbJEKGBom7OXfNyy7sWILz3nxRAE,6545
-trainml/cli/job/create.py,sha256=MDf53CFua58uJoW4n9FZT1XERJOvAYunEc8tcuO6WQ8,35098
+trainml/cli/job/__init__.py,sha256=LwnagzsL517w1dLZ9BqEd-ujAtlQaRfeZYMlXsI5qYw,7429
+trainml/cli/job/create.py,sha256=DegZYOggCNOSCUzeMTmpSrQRSMlYrUSV4XcElaocf5g,35157
 trainml/cli/project/__init__.py,sha256=HDcJUbKMHhz4Thrvpst5hnywFqzsv0XWmvfKNRi8zuo,1918
 trainml/cli/project/credential.py,sha256=gByXKiYf5sJeNRtuXWcercWv8P2IzO5TjT8Ypp4mCR8,3443
 trainml/cli/project/data_connector.py,sha256=KLDhpNJfwcJkcmyuJRgcVH70Jf73619O9ddYP8vhMvk,1411
@@ -133,9 +136,12 @@ trainml/projects/members.py,sha256=7fptZ2leIQEUiRtvd3z15RTOJyPafzc5wwQLNxICElQ,2
 trainml/projects/projects.py,sha256=0GIKLV1ZTP_LY9bvXKwiDe_I10YkVAhDUlc4AbdDFVM,2840
 trainml/projects/secrets.py,sha256=TIvBd3rAvd4lF3pm5qR98UslHjldzlnzn_n9yvpmLgg,2160
 trainml/projects/services.py,sha256=rI-uFmojqOTNLbqBeX6gaSwMkI6LKzRuJthQCH0A2h4,2771
-trainml-0.5.17.dist-info/LICENSE,sha256=s0lpBxhSSUEpMavwde-Vb6K_K7xDCTTvSpNznVqVGR0,1069
-trainml-0.5.17.dist-info/METADATA,sha256=MQaZfpWnJ7BtkOW4AEAk28YuwWgcpIrsTJgr0G1buLE,7346
-trainml-0.5.17.dist-info/WHEEL,sha256=pkctZYzUS4AYVn6dJ-7367OJZivF2e8RA9b_ZBjif18,92
-trainml-0.5.17.dist-info/entry_points.txt,sha256=OzBDm2wXby1bSGF02jTVxzRFZLejnbFiLHXhKdW3Bds,63
-trainml-0.5.17.dist-info/top_level.txt,sha256=Y1kLFRWKUW7RG8BX7cvejHF_yW8wBOaRYF1JQHENY4w,23
-trainml-0.5.17.dist-info/RECORD,,
+trainml/utils/__init__.py,sha256=YGZaaZGxeZ1Lz8ScoIx_qCDbNHsvmNORJTxC8Bxd_ro,39
+trainml/utils/auth.py,sha256=v28FItj3zj5nrfXBt2GM5Av1y-1Ug5G39ejS5wrWh0U,26573
+trainml/utils/transfer.py,sha256=TcoHQdeRoDRdGj1B6VCjwW43tHHnBpP-zl__5t7pCz4,25118
+trainml-1.0.1.dist-info/LICENSE,sha256=s0lpBxhSSUEpMavwde-Vb6K_K7xDCTTvSpNznVqVGR0,1069
+trainml-1.0.1.dist-info/METADATA,sha256=fGhhOAR13XSyoKTyh3TQYJ6pzYEgg4EoYbM3D3EUr18,7288
+trainml-1.0.1.dist-info/WHEEL,sha256=pkctZYzUS4AYVn6dJ-7367OJZivF2e8RA9b_ZBjif18,92
+trainml-1.0.1.dist-info/entry_points.txt,sha256=OzBDm2wXby1bSGF02jTVxzRFZLejnbFiLHXhKdW3Bds,63
+trainml-1.0.1.dist-info/top_level.txt,sha256=Y1kLFRWKUW7RG8BX7cvejHF_yW8wBOaRYF1JQHENY4w,23
+trainml-1.0.1.dist-info/RECORD,,