trainml 0.5.15__tar.gz → 0.5.17__tar.gz

{trainml-0.5.15 → trainml-0.5.17}/PKG-INFO

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

{trainml-0.5.15 → trainml-0.5.17}/tests/unit/projects/test_projects_unit.py

@@ -75,7 +75,7 @@ class ProjectsTests:
         requested_config = dict(
             name="new project",
         )
-        expected_payload = dict(name="new project", copy_credentials=False)
+        expected_payload = dict(name="new project")
         api_response = {
             "id": "project-id-1",
             "name": "new project",

{trainml-0.5.15 → trainml-0.5.17}/trainml/__init__.py

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

{trainml-0.5.15 → trainml-0.5.17}/trainml/cloudbender/services.py

@@ -2,6 +2,7 @@ import json
 import logging
 import asyncio
 import math
+from enum import Enum
 
 from trainml.exceptions import (
     ApiError,
@@ -9,6 +10,14 @@ from trainml.exceptions import (
     TrainMLException,
 )
 
+class SERVICE_CERT_ALGORITHMS(str, Enum):
+    RSA_2048 = 'rsa2048'
+    RSA_4096 = 'rsa4096'
+    ED25519 = 'ed25519'
+    P256 = 'p256'
+    P384 ='p384'
+    P521='p521'
+
 
 class Services(object):
     def __init__(self, trainml):
@@ -178,11 +187,12 @@ class Service:
 
         raise TrainMLException(f"Timeout waiting for {status}")
     
-    async def generate_certificate(self, **kwargs):
+    async def generate_certificate(self, algorithm=SERVICE_CERT_ALGORITHMS.ED25519.value, **kwargs):
         resp = await self.trainml._query(
             f"/provider/{self._provider_uuid}/region/{self._region_uuid}/service/{self._id}/certificate",
             "POST",
-            kwargs
+            kwargs,
+            dict(algorithm=algorithm)
         )
         self.__init__(self.trainml, **resp)
         return self

{trainml-0.5.15 → trainml-0.5.17}/trainml/projects/projects.py

@@ -27,10 +27,10 @@ class Projects(object):
         projects = [Project(self.trainml, **project) for project in resp]
         return projects
 
-    async def create(self, name, copy_credentials=False, **kwargs):
+    async def create(self, name, **kwargs):
         data = dict(
             name=name,
-            copy_credentials=copy_credentials,
+            **kwargs
         )
         payload = {k: v for k, v in data.items() if v is not None}
         logging.info(f"Creating Project {name}")

{trainml-0.5.15 → trainml-0.5.17}/trainml.egg-info/PKG-INFO

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