Skip to content

Commit

Permalink
read existing custom tags from databricks config template (#555)
Browse files Browse the repository at this point in the history
  • Loading branch information
esadler-hbo authored Aug 7, 2022
1 parent 289e329 commit a2e86e8
Showing 1 changed file with 8 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -149,7 +149,14 @@ def submit_feathr_job(self, job_name: str, main_jar_path: str, main_class_name:
configuration['spark.executor.extraJavaOptions'] = '-Djava.security.properties='
configuration['spark.driver.extraJavaOptions'] = '-Djava.security.properties='
submission_params['new_cluster']['spark_conf'] = configuration
submission_params['new_cluster']['custom_tags'] = job_tags

if job_tags:
custom_tags = submission_params['new_cluster'].get('custom_tags', {})
for tag, value in job_tags.items():
custom_tags[tag] = value

submission_params['new_cluster']['custom_tags'] = custom_tags

# the feathr main jar file is anyway needed regardless it's pyspark or scala spark
if not main_jar_path:
logger.info(f"Main JAR file is not set, using default package '{FEATHR_MAVEN_ARTIFACT}' from Maven")
Expand Down

0 comments on commit a2e86e8

Please sign in to comment.