ellipsis 3.1.51__tar.gz → 3.1.52__tar.gz

{ellipsis-3.1.51 → ellipsis-3.1.52}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ellipsis
-Version: 3.1.51
+Version: 3.1.52
 Summary: Package to interact with the Ellipsis API
 Home-page: https://github.com/ellipsis-drive-internal/python-package
 Author: Daniel van der Maas

{ellipsis-3.1.51 → ellipsis-3.1.52}/ellipsis/__init__.py

@@ -6,7 +6,7 @@ from ellipsis import user
 from ellipsis import path
 from ellipsis import util
 from ellipsis import compute
-__version__ = '3.1.51'
+__version__ = '3.1.52'
 
 
 

{ellipsis-3.1.51 → ellipsis-3.1.52}/ellipsis/compute/root.py

@@ -7,10 +7,9 @@ from ellipsis import sanitize
 from ellipsis import apiManager
 from ellipsis.account import getInfo
 
-
+apiManager.baseUrl = 'https://acc.api.ellipsis-drive.com/v3'
 
 def createCluster(layers, token, nodes=None, interpreter='python3.12', requirements= [], awaitTillStarted = True):
-
     layers = sanitize.validDictArray('layers', layers, True)
     token = sanitize.validString('token', token, True)
     nodes = sanitize.validInt('nodes', nodes, False)
@@ -32,9 +31,11 @@ def createCluster(layers, token, nodes=None, interpreter='python3.12', requireme
     while awaitTillStarted:
         res = listClusters(token=token)['result']
         r = [x for x in res if x['id'] == clusterId][0]
-
         if r['status'] == 'available':
             break
+        if r['status'] == 'errored':
+            raise ValueError(r['message'])
+
         time.sleep(1)
 
     return {'id':clusterId}
@@ -60,7 +61,12 @@ def execute(clusterId, f, token, awaitTillCompleted=True):
             break
         time.sleep(1)
 
-    return r['result']
+    for x in r['result']:
+        if x['type'] == 'exception':
+            raise x['value']
+
+    values = [x['value'] for x in r['result']]
+    return values
 
 def parseResults(r):
     results = []
@@ -74,6 +80,7 @@ def parseResults(r):
 def terminateCluster(clusterId, token, awaitTillTerminated = True):
     clusterId = sanitize.validUuid('clusterId', clusterId, True)
     token = sanitize.validString('token', token, True)
+    sanitize.validBool('awaitTillTerminated',awaitTillTerminated, True)
 
 
     r = apiManager.post('/compute/' + clusterId + '/terminate', {}, token)
@@ -87,6 +94,26 @@ def terminateCluster(clusterId, token, awaitTillTerminated = True):
 
     return r
 
+def terminateAll(token, awaitTillTerminated = True ):
+    token = sanitize.validString('token', token, True)
+    sanitize.validBool('awaitTillTerminated',awaitTillTerminated, True)
+
+    res = listClusters(token = token)['result']
+
+    for x in res:
+        if x['status'] != 'stopped' and x['status'] != 'errored' and x['status'] != 'stopping':
+
+            apiManager.post('/compute/' + x['id'] + '/terminate', {}, token)
+
+            while awaitTillTerminated:
+                res = listClusters(token=token)['result']
+                z = [x for x in res if x['id'] == x['id']][0]
+                if z['status'] == 'stopped':
+                    break
+                time.sleep(1)
+
+
+
 def getClusterInfo(clusterId, token):
     res = listClusters(token=token)['result']
     r = [x for x in res if x['id'] == clusterId]

{ellipsis-3.1.51 → ellipsis-3.1.52}/ellipsis.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ellipsis
-Version: 3.1.51
+Version: 3.1.52
 Summary: Package to interact with the Ellipsis API
 Home-page: https://github.com/ellipsis-drive-internal/python-package
 Author: Daniel van der Maas

{ellipsis-3.1.51 → ellipsis-3.1.52}/ellipsis.egg-info/requires.txt

@@ -1,6 +1,7 @@
 Fiona
 Pillow
 Shapely
+dill
 geopandas
 geopy
 imagecodecs

{ellipsis-3.1.51 → ellipsis-3.1.52}/setup.py

@@ -6,7 +6,7 @@ with open("README.md", "r") as fh:
 
 setuptools.setup(
     name="ellipsis",
-    version="3.1.51",
+    version="3.1.52",
     author="Daniel van der Maas",
     author_email="daniel@ellipsis-drive.com",
     description="Package to interact with the Ellipsis API",
@@ -36,7 +36,8 @@ setuptools.setup(
     'scikit-image',
     'Fiona',
     'pygltflib',
-    'tifffile'
+    'tifffile',
+    'dill'
     ],
     python_requires='>=3.6',
 )