anatools 5.1.27__tar.gz → 5.1.28__tar.gz

{anatools-5.1.27/anatools.egg-info → anatools-5.1.28}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: anatools
-Version: 5.1.27
+Version: 5.1.28
 Summary: Tools for development with the Rendered.ai Platform.
 Home-page: https://rendered.ai
 Author: Rendered AI, Inc

{anatools-5.1.27 → anatools-5.1.28}/anatools/__init__.py

@@ -16,4 +16,4 @@ View the `Introduction to Rendered.ai Documentation`_ to learn more.
 from .anaclient.anaclient import client
 from .annotations.annotations import annotations
 
-__version__ = '5.1.27'
+__version__ = '5.1.28'

{anatools-5.1.27 → anatools-5.1.28}/anatools/anaclient/api/datasets.py

@@ -53,7 +53,7 @@ def getDatasetJobs(self, organizationId, workspaceId, datasetId=None, cursor=Non
     return self.errorhandler(response, "getDatasetJobs")
 
 
-def createDataset(self, workspaceId, graphId, name, description=None, runs=1, seed=0, priority=1, tags=[]):
+def createDataset(self, workspaceId, graphId, name, description=None, runs=1, seed=0, priority=1, compressDataset=True, tags=[]):
     response = self.session.post(
         url = self.url, 
         headers = self.headers, 
@@ -67,11 +67,12 @@ def createDataset(self, workspaceId, graphId, name, description=None, runs=1, se
                 "runs": runs,
                 "seed": seed,
                 "priority": priority,
+                "compressDataset": compressDataset,
                 "tags": tags
             },
             "query": """mutation 
-                createDataset($workspaceId: String!, $graphId: String!, $name: String!, $description: String, $runs: Int!, $seed: Int!, $priority: Int!, $tags: [String]) {
-                    createDataset(workspaceId: $workspaceId, graphId: $graphId, name: $name, description: $description, runs: $runs, seed: $seed, priority: $priority, tags: $tags)
+                createDataset($workspaceId: String!, $graphId: String!, $name: String!, $description: String, $runs: Int!, $seed: Int!, $priority: Int!, $compressDataset: Boolean, $tags: [String]) {
+                    createDataset(workspaceId: $workspaceId, graphId: $graphId, name: $name, description: $description, runs: $runs, seed: $seed, priority: $priority, compressDataset: $compressDataset, tags: $tags)
                 }"""})
     return self.errorhandler(response, "createDataset")
 

{anatools-5.1.27 → anatools-5.1.28}/anatools/anaclient/datasets.py

@@ -79,7 +79,7 @@ def get_dataset_jobs(self, organizationId=None, workspaceId=None, datasetId=None
     return jobs
 
 
-def create_dataset(self, name, graphId, description='', runs=1, priority=1, seed=1, tags=[], workspaceId=None):
+def create_dataset(self, name, graphId, description='', runs=1, priority=1, seed=1, compressDataset=True, tags=[], workspaceId=None):
     """Create a new synthetic dataset using a graph in the workspace. This will start a new dataset job in the workspace.
     
     Parameters
@@ -96,6 +96,8 @@ def create_dataset(self, name, graphId, description='', runs=1, priority=1, seed
         Job priority.
     seed : int
         Seed number.
+    compressDataset : bool
+        Whether to compress the dataset. If false, the dataset asset will be accessed through mount_workspaces.
     tags : list[str]
         Tags for new dataset.
     workspaceId : str
@@ -111,7 +113,7 @@ def create_dataset(self, name, graphId, description='', runs=1, priority=1, seed
     if graphId is None: raise ValueError("The graphId parameter is required!")
     if description is None: description = ''
     if workspaceId is None: workspaceId = self.workspace
-    return self.ana_api.createDataset(workspaceId=workspaceId, graphId=graphId, name=name, description=description, runs=runs, seed=seed, priority=priority, tags=tags)
+    return self.ana_api.createDataset(workspaceId=workspaceId, graphId=graphId, name=name, description=description, runs=runs, seed=seed, priority=priority, compressDataset=compressDataset, tags=tags)
 
 
 def edit_dataset(self, datasetId, description=None, name=None, pause=None, priority=None, tags=None, workspaceId=None):

{anatools-5.1.27 → anatools-5.1.28/anatools.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: anatools
-Version: 5.1.27
+Version: 5.1.28
 Summary: Tools for development with the Rendered.ai Platform.
 Home-page: https://rendered.ai
 Author: Rendered AI, Inc