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

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/__init__.py

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

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/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.dist-info → trainml-0.5.17.dist-info}/METADATA

@@ -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.dist-info → trainml-0.5.17.dist-info}/RECORD

@@ -77,8 +77,8 @@ tests/unit/projects/test_project_datastores_unit.py,sha256=kcoaSs54gGqJ9gwPMpP6K
 tests/unit/projects/test_project_members_unit.py,sha256=C5N6jzT7zM3eL3_bXEddddBHPFJ3617T0lYluBjMy78,3466
 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=8rJ40bJ7YszEsoL6D02Muey-sgRjqFmMoR-R3IiP_Ig,3810
-trainml/__init__.py,sha256=yxX-2bQsXFk3KgMV8b7day_Wz6qy6Ag4kOHvcMCv15k,433
+tests/unit/projects/test_projects_unit.py,sha256=S4ydA9JLgGzFwaY4b30YJF0Pl6Iog4ZdzQmHpLy_UEA,3786
+trainml/__init__.py,sha256=an9i7gL01ZrVyaS0RF03wAeRJuWpgA1MgE8-tcwrH1c,433
 trainml/__main__.py,sha256=JgErYkiskih8Y6oRwowALtR-rwQhAAdqOYWjQraRIPI,59
 trainml/auth.py,sha256=clbx5S5prJ3u62aEESdBXIHF_HmreQ-L1ShxcyWQNDs,26565
 trainml/checkpoints.py,sha256=Hg3_Si7ohzsv4_8JIjoj9pMRpHNZRWKC5R2gtgmp01s,8790
@@ -124,18 +124,18 @@ trainml/cloudbender/devices.py,sha256=QORNmKdLJoqGZmeWXRnivC1JmNBIw-ebvf4bsoem3r
 trainml/cloudbender/nodes.py,sha256=U2sTIL2fuBVsNFPsJrvB2JbBcuLULF1-AwJ4dp5ChmM,5924
 trainml/cloudbender/providers.py,sha256=22ymUl1KLC8UlyoWMsIrOKhUDOwnhklhHQ3EZ_V6eWA,3738
 trainml/cloudbender/regions.py,sha256=nfSY9fIWp_AaRE_1Y0qwXX6WVSyPKxpji-zUfM3BNUo,5157
-trainml/cloudbender/services.py,sha256=bJtp9ljq8g10r-04phV_XXS8mS8YR3KMbJ4XtHa2SOI,5773
+trainml/cloudbender/services.py,sha256=j--8V_JtzhOrWDFXcuyQCZoSdxYKaNtpxTV9qTsva5E,6051
 trainml/projects/__init__.py,sha256=6NKCcHtQMeGB1IyU-djANphfnDX6MEkrXUM5Fyq9fWg,75
 trainml/projects/credentials.py,sha256=6WqHy_-SZZwqE4rULLF8gSyeRVmfsUxqZBuCjBXyxKw,2255
 trainml/projects/data_connectors.py,sha256=WZATdUq4vE3x47Ny5HDwPD7lIUx0syjaSE9BmFWNuEg,2216
 trainml/projects/datastores.py,sha256=qocqD5mGfm2V_wh36CEh3oldnIeJg57ppc6CM129Dkk,2095
 trainml/projects/members.py,sha256=7fptZ2leIQEUiRtvd3z15RTOJyPafzc5wwQLNxICElQ,2826
-trainml/projects/projects.py,sha256=5_6HUg9ZYq8v69QPgC4Yfd_DM0OlZd8tnxh2ahjOgm0,2890
+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.15.dist-info/LICENSE,sha256=s0lpBxhSSUEpMavwde-Vb6K_K7xDCTTvSpNznVqVGR0,1069
-trainml-0.5.15.dist-info/METADATA,sha256=kFbUgFw2pcOjJgySdhnKmVrkVpkFMI6mtVrN7t26yXA,7346
-trainml-0.5.15.dist-info/WHEEL,sha256=pkctZYzUS4AYVn6dJ-7367OJZivF2e8RA9b_ZBjif18,92
-trainml-0.5.15.dist-info/entry_points.txt,sha256=OzBDm2wXby1bSGF02jTVxzRFZLejnbFiLHXhKdW3Bds,63
-trainml-0.5.15.dist-info/top_level.txt,sha256=Y1kLFRWKUW7RG8BX7cvejHF_yW8wBOaRYF1JQHENY4w,23
-trainml-0.5.15.dist-info/RECORD,,
+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,,