trainml 0.5.2__tar.gz → 0.5.4__tar.gz

{trainml-0.5.2/trainml.egg-info → trainml-0.5.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: trainml
-Version: 0.5.2
+Version: 0.5.4
 Summary: trainML client SDK and command line utilities
 Home-page: https://github.com/trainML/trainml-cli
 Author: trainML

{trainml-0.5.2 → trainml-0.5.4}/tests/integration/test_jobs_integration.py

@@ -430,7 +430,7 @@ class JobIOTests:
             source_uri="git@github.com:trainML/environment-tests.git",
         )
         await model.wait_for("ready", 300)
-        assert model.size >= 200000
+        assert model.size >= 50000
 
         job = await trainml.jobs.create(
             "CLI Automated Tests - Training With trainML Model Output",

{trainml-0.5.2 → trainml-0.5.4}/trainml/__init__.py

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

{trainml-0.5.2 → trainml-0.5.4}/trainml/cli/job/create.py

@@ -268,9 +268,7 @@ def notebook(
         options["environment"]["type"] = environment
 
     try:
-        envs = [
-            {"key": e.split("=")[0], "value": e.split("=")[1]} for e in env
-        ]
+        envs = [{"key": e.split("=")[0], "value": e.split("=")[1]} for e in env]
         options["environment"]["env"] = envs
     except IndexError:
         raise click.UsageError(
@@ -284,16 +282,12 @@ def notebook(
         if pip_packages:
             options["environment"]["packages"]["pip"] = pip_packages.split(",")
         if conda_packages:
-            options["environment"]["packages"]["conda"] = conda_packages.split(
-                ","
-            )
+            options["environment"]["packages"]["conda"] = conda_packages.split(",")
 
     if data_dir:
         click.echo("Creating Dataset..", file=config.stdout)
         new_dataset = config.trainml.run(
-            config.trainml.client.datasets.create(
-                f"Job - {name}", "local", data_dir
-            )
+            config.trainml.client.datasets.create(f"Job - {name}", "local", data_dir)
         )
         if attach:
             config.trainml.run(new_dataset.attach(), new_dataset.connect())
@@ -302,9 +296,7 @@ def notebook(
             config.trainml.run(new_dataset.connect())
             config.trainml.run(new_dataset.wait_for("ready"))
             config.trainml.run(new_dataset.disconnect())
-        options["data"]["datasets"].append(
-            dict(id=new_dataset.id, type="existing")
-        )
+        options["data"]["datasets"].append(dict(id=new_dataset.id, type="existing"))
 
     if git_uri:
         options["model"]["source_type"] = "git"
@@ -397,7 +389,8 @@ def notebook(
         ],
         case_sensitive=False,
     ),
-    default="rtx3090",
+    default=["rtx3090"],
+    multiple=True,
     show_default=True,
     help="GPU type.",
 )
@@ -619,21 +612,15 @@ def training(
     if output_type:
         options["data"]["output_type"] = output_type
         options["data"]["output_uri"] = output_uri
-        options["data"]["output_options"] = dict(
-            archive=archive, save_model=save_model
-        )
+        options["data"]["output_options"] = dict(archive=archive, save_model=save_model)
 
     if output_dir:
         options["data"]["output_type"] = "local"
         options["data"]["output_uri"] = output_dir
-        options["data"]["output_options"] = dict(
-            archive=archive, save_model=save_model
-        )
+        options["data"]["output_options"] = dict(archive=archive, save_model=save_model)
 
     try:
-        envs = [
-            {"key": e.split("=")[0], "value": e.split("=")[1]} for e in env
-        ]
+        envs = [{"key": e.split("=")[0], "value": e.split("=")[1]} for e in env]
         options["environment"]["env"] = envs
     except IndexError:
         raise click.UsageError(
@@ -647,16 +634,12 @@ def training(
         if pip_packages:
             options["environment"]["packages"]["pip"] = pip_packages.split(",")
         if conda_packages:
-            options["environment"]["packages"]["conda"] = conda_packages.split(
-                ","
-            )
+            options["environment"]["packages"]["conda"] = conda_packages.split(",")
 
     if data_dir:
         click.echo("Creating Dataset..", file=config.stdout)
         new_dataset = config.trainml.run(
-            config.trainml.client.datasets.create(
-                f"Job - {name}", "local", data_dir
-            )
+            config.trainml.client.datasets.create(f"Job - {name}", "local", data_dir)
         )
         if attach:
             config.trainml.run(new_dataset.attach(), new_dataset.connect())
@@ -665,9 +648,7 @@ def training(
             config.trainml.run(new_dataset.connect())
             config.trainml.run(new_dataset.wait_for("ready"))
             config.trainml.run(new_dataset.disconnect())
-        options["data"]["datasets"].append(
-            dict(id=new_dataset.id, type="existing")
-        )
+        options["data"]["datasets"].append(dict(id=new_dataset.id, type="existing"))
 
     if git_uri:
         options["model"]["source_type"] = "git"
@@ -751,8 +732,9 @@ def training(
         ],
         case_sensitive=False,
     ),
-    default="rtx3090",
+    default=["rtx3090"],
     show_default=True,
+    multiple=True,
     help="GPU type.",
 )
 @click.option(
@@ -975,21 +957,15 @@ def inference(
     if output_type:
         options["data"]["output_type"] = output_type
         options["data"]["output_uri"] = output_uri
-        options["data"]["output_options"] = dict(
-            archive=archive, save_model=save_model
-        )
+        options["data"]["output_options"] = dict(archive=archive, save_model=save_model)
 
     if output_dir:
         options["data"]["output_type"] = "local"
         options["data"]["output_uri"] = output_dir
-        options["data"]["output_options"] = dict(
-            archive=archive, save_model=save_model
-        )
+        options["data"]["output_options"] = dict(archive=archive, save_model=save_model)
 
     try:
-        envs = [
-            {"key": e.split("=")[0], "value": e.split("=")[1]} for e in env
-        ]
+        envs = [{"key": e.split("=")[0], "value": e.split("=")[1]} for e in env]
         options["environment"]["env"] = envs
     except IndexError:
         raise click.UsageError(
@@ -1003,9 +979,7 @@ def inference(
         if pip_packages:
             options["environment"]["packages"]["pip"] = pip_packages.split(",")
         if conda_packages:
-            options["environment"]["packages"]["conda"] = conda_packages.split(
-                ","
-            )
+            options["environment"]["packages"]["conda"] = conda_packages.split(",")
 
     if git_uri:
         options["model"]["source_type"] = "git"
@@ -1125,8 +1099,9 @@ def from_json(config, attach, connect, file):
         ],
         case_sensitive=False,
     ),
-    default="rtx3090",
+    default=["rtx3090"],
     show_default=True,
+    multiple=True,
     help="GPU type.",
 )
 @click.option(
@@ -1296,9 +1271,7 @@ def endpoint(
         options["environment"]["type"] = environment
 
     try:
-        envs = [
-            {"key": e.split("=")[0], "value": e.split("=")[1]} for e in env
-        ]
+        envs = [{"key": e.split("=")[0], "value": e.split("=")[1]} for e in env]
         options["environment"]["env"] = envs
     except IndexError:
         raise click.UsageError(
@@ -1312,9 +1285,7 @@ def endpoint(
         if pip_packages:
             options["environment"]["packages"]["pip"] = pip_packages.split(",")
         if conda_packages:
-            options["environment"]["packages"]["conda"] = conda_packages.split(
-                ","
-            )
+            options["environment"]["packages"]["conda"] = conda_packages.split(",")
 
     if git_uri:
         options["model"]["source_type"] = "git"
@@ -1356,6 +1327,4 @@ def endpoint(
             click.echo("Waiting for job to start...", file=config.stdout)
             config.trainml.run(job.wait_for("running", timeout))
             config.trainml.run(job.refresh())
-            click.echo(
-                f"Endpoint is running at:  {job.url}", file=config.stdout
-            )
+            click.echo(f"Endpoint is running at:  {job.url}", file=config.stdout)

{trainml-0.5.2 → trainml-0.5.4/trainml.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: trainml
-Version: 0.5.2
+Version: 0.5.4
 Summary: trainML client SDK and command line utilities
 Home-page: https://github.com/trainML/trainml-cli
 Author: trainML