update: add owner_id to create

Author: VsevolodX

exabyte_api_client/endpoints/entity.py

@@ -76,16 +76,19 @@ def update(self, id_, modifier):
         """
         return self.request("PATCH", "/".join((self.name, id_)), data=json.dumps(modifier), headers=self.headers)
 
-    def create(self, config):
+    def create(self, config, owner_id=None):
         """
         Creates a new entity.
 
         Args:
             config (dict): entity config.
+            owner_id (str): owner ID. Entity is created under user's default account if not specified.
 
         Returns:
              dict: new entity.
         """
+        if owner_id:
+            config["owner"] = {"_id": owner_id}
         return self.request("PUT", "/".join((self.name, "create")), data=json.dumps(config), headers=self.headers)
 
     def copy(self, id_):

exabyte_api_client/endpoints/jobs.py

@@ -110,20 +110,20 @@ def get_compute(self, cluster, ppn=1, nodes=1, queue="D", time_limit="01:00:00",
             "arguments": {},
         }
 
-    def create_by_ids(self, materials, workflow_id, project_id, owner_id, prefix, compute=None):
+    def create_by_ids(self, materials, workflow_id, project_id, prefix, owner_id=None, compute=None):
         """
         Creates jobs from the given materials
 
         Args:
             materials (list[dict]): list of materials.
             workflow_id (str): workflow ID.
             project_id (str): project ID.
-            owner_id (str): owner ID.
-            compute (dict): compute configuration.
             prefix (str): job prefix.
+            owner_id (str, optional): owner ID.
+            compute (dict, optional): compute configuration.
 
         Returns:
-            list
+            list: List of created jobs.
         """
         jobs = []
         for material in materials: