Skip to content

Commit

Permalink
fix lint errors
Browse files Browse the repository at this point in the history
  • Loading branch information
santoshk_qubole committed Mar 30, 2020
1 parent 735f24d commit 6218b9e
Showing 1 changed file with 10 additions and 5 deletions.
15 changes: 10 additions & 5 deletions qds_sdk/engine.py
Original file line number Diff line number Diff line change
Expand Up @@ -78,11 +78,16 @@ def set_engine_config(self,
default_pool, enable_rubix)
self.set_presto_settings(presto_version, custom_presto_config)
self.set_spark_settings(spark_version, custom_spark_config)
# because of same config dbtap_id used in mlflow and airflow, configs are getting created for both cluster type
# because of same config dbtap_id used in mlflow and airflow,
# configs are getting created for both cluster type
if mlflow_version:
self.set_mlflow_settings(mlflow_version, dbtap_id)
if airflow_version:
self.set_airflow_settings(dbtap_id, fernet_key, overrides, airflow_version, airflow_python_version)
self.set_airflow_settings(dbtap_id,
fernet_key,
overrides,
airflow_version,
airflow_python_version)

def set_fairscheduler_settings(self,
fairscheduler_config_xml=None,
Expand Down Expand Up @@ -259,7 +264,7 @@ def engine_parser(argparser):
default=None,
help="mlflow version for mlflow cluster", )
mlflow_settings_group.add_argument("--mlflow-dbtap-id",
dest="dbtap_id",
default=None,
help="dbtap id for mlflow cluster", )
dest="dbtap_id",
default=None,
help="dbtap id for mlflow cluster", )

0 comments on commit 6218b9e

Please sign in to comment.