Added support for overriding defined tags in ads job run. (#207)

Author: lu-ohai

ads/jobs/ads_job.py

@@ -383,7 +383,14 @@ def create(self, **kwargs) -> "Job":
         return self
 
     def run(
-        self, name=None, args=None, env_var=None, freeform_tags=None, wait=False
+        self,
+        name=None,
+        args=None,
+        env_var=None,
+        freeform_tags=None,
+        defined_tags=None,
+        wait=False,
+        **kwargs
     ) -> Union[DataScienceJobRun, DataFlowRun]:
         """Runs the job.
 
@@ -404,11 +411,15 @@ def run(
             Additional environment variables for the job run, by default None
         freeform_tags : dict, optional
             Freeform tags for the job run, by default None
+        defined_tags : dict, optional
+            Defined tags for the job run, by default None
         wait : bool, optional
             Indicate if this method call should wait for the job run.
             By default False, this method returns as soon as the job run is created.
             If this is set to True, this method will stream the job logs and wait until it finishes,
             similar to `job.run().watch()`.
+        kwargs
+            additional keyword arguments
 
         Returns
         -------
@@ -423,7 +434,8 @@ def run(
                 name="<my_job_run_name>",
                 args="new_arg --new_key new_val",
                 env_var={"new_env": "new_val"},
-                freeform_tags={"new_tag": "new_tag_val"}
+                freeform_tags={"new_tag": "new_tag_val"},
+                defined_tags={"Operations": {"CostCenter": "42"}}
             )
 
         """
@@ -432,7 +444,9 @@ def run(
             args=args,
             env_var=env_var,
             freeform_tags=freeform_tags,
+            defined_tags=defined_tags,
             wait=wait,
+            **kwargs
         )
 
     def run_list(self, **kwargs) -> list:

ads/jobs/builders/infrastructure/dsc_job.py

@@ -1542,6 +1542,7 @@ def run(
         freeform_tags=None,
         defined_tags=None,
         wait=False,
+        **kwargs
     ) -> DataScienceJobRun:
         """Runs a job on OCI Data Science job
 
@@ -1559,6 +1560,8 @@ def run(
             Defined tags for the job run, by default None
         wait : bool, optional
             Indicate if this method should wait for the run to finish before it returns, by default False.
+        kwargs
+            additional keyword arguments
 
         Returns
         -------
@@ -1597,6 +1600,7 @@ def run(
             freeform_tags=freeform_tags,
             defined_tags=defined_tags,
             wait=wait,
+            **kwargs
         )
 
     def delete(self) -> None: