proximl 0.5.1__tar.gz → 0.5.2__tar.gz

{proximl-0.5.1/proximl.egg-info → proximl-0.5.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: proximl
-Version: 0.5.1
+Version: 0.5.2
 Summary: proxiML client SDK and command line utilities
 Home-page: https://github.com/proxiML/python-sdk
 Author: proxiML

{proximl-0.5.1 → proximl-0.5.2}/proximl/__init__.py

@@ -13,5 +13,5 @@ logging.basicConfig(
 logger = logging.getLogger(__name__)
 
 
-__version__ = "0.5.1"
+__version__ = "0.5.2"
 __all__ = "ProxiML"

{proximl-0.5.1 → proximl-0.5.2}/proximl/cli/dataset.py

@@ -252,9 +252,7 @@ def rename(config, dataset, name):
     DATASET may be specified by name or ID, but ID is preferred.
     """
     try:
-        dataset = config.proximl.run(
-            config.proximl.client.datasets.get(dataset)
-        )
+        dataset = config.proximl.run(config.proximl.client.datasets.get(dataset))
         if dataset is None:
             raise click.UsageError("Cannot find specified dataset.")
     except:

{proximl-0.5.1 → proximl-0.5.2}/proximl/cli/job/create.py

@@ -683,7 +683,7 @@ def training(
         config.proximl.client.jobs.create(
             name=name,
             type="training",
-            gpu_type=gpu_type,
+            gpu_types=gpu_type,
             cpu_count=cpu_count,
             disk_size=disk_size,
             workers=[command for command in commands],
@@ -1021,7 +1021,7 @@ def inference(
         config.proximl.client.jobs.create(
             name=name,
             type="inference",
-            gpu_type=gpu_type,
+            gpu_types=gpu_type,
             cpu_count=cpu_count,
             disk_size=disk_size,
             workers=[command],
@@ -1330,7 +1330,7 @@ def endpoint(
         config.proximl.client.jobs.create(
             name=name,
             type="endpoint",
-            gpu_type=gpu_type,
+            gpu_types=gpu_type,
             cpu_count=cpu_count,
             disk_size=disk_size,
             endpoint=dict(routes=routes),

{proximl-0.5.1 → proximl-0.5.2}/proximl/datasets.py

@@ -37,8 +37,7 @@ class Datasets(object):
             source_type=source_type,
             source_uri=source_uri,
             source_options=kwargs.get("source_options"),
-            project_uuid=kwargs.get("project_uuid")
-            or self.proximl.active_project,
+            project_uuid=kwargs.get("project_uuid") or self.proximl.active_project,
         )
         payload = {k: v for k, v in data.items() if v is not None}
         logging.info(f"Creating Dataset {name}")
@@ -119,9 +118,7 @@ class Dataset:
                 entity_type="dataset",
                 project_uuid=self._dataset.get("project_uuid"),
                 cidr=self._dataset.get("vpn").get("cidr"),
-                ssh_port=self._dataset.get("vpn")
-                .get("client")
-                .get("ssh_port"),
+                ssh_port=self._dataset.get("vpn").get("client").get("ssh_port"),
                 input_path=self._dataset.get("source_uri")
                 if self.status in ["new", "downloading"]
                 else None,
@@ -191,9 +188,7 @@ class Dataset:
                 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()}"
                     )
@@ -230,9 +225,7 @@ class Dataset:
             return
         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:

{proximl-0.5.1 → proximl-0.5.2/proximl.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: proximl
-Version: 0.5.1
+Version: 0.5.2
 Summary: proxiML client SDK and command line utilities
 Home-page: https://github.com/proxiML/python-sdk
 Author: proxiML

{proximl-0.5.1 → proximl-0.5.2}/tests/integration/test_jobs_integration.py

@@ -191,8 +191,7 @@ class JobAPIResourceValidationTests:
                 disk_size=10,
             )
         assert (
-            "Invalid Request - CPU Count must be a multiple of 4"
-            in error.value.message
+            "Invalid Request - CPU Count must be a multiple of 4" in error.value.message
         )
 
     async def test_invalid_gpu_count_for_cpu(self, proximl):
@@ -431,7 +430,7 @@ class JobIOTests:
             source_uri="git@github.com:proxiML/environment-tests.git",
         )
         await model.wait_for("ready", 300)
-        assert model.size >= 500000
+        assert model.size >= 200000
 
         job = await proximl.jobs.create(
             "CLI Automated Tests - Training With proxiML Model Output",
@@ -588,10 +587,7 @@ class JobTypeTests:
         sys.stderr.write(captured.err)
         assert "Epoch 1/2" in captured.out
         assert "Epoch 2/2" in captured.out
-        assert (
-            "Uploading s3://proximl-example/output/resnet_cifar10"
-            in captured.out
-        )
+        assert "Uploading s3://proximl-example/output/resnet_cifar10" in captured.out
         assert (
             "upload: ./model.ckpt-0002.data-00000-of-00001 to s3://proximl-example/output/resnet_cifar10/model.ckpt-0002.data-00000-of-00001"
             in captured.out