Support YARN_CONF_DIR along with HADOOP_CONF_DIR (#101)

ei-grad · web-flow · commit ca2a3d14f5d7 · 2021-07-15T13:52:51.000-07:00
When both defined then YARN_CONF_DIR should have priority as it is more
specific.
diff --git a/yarn_api_client/hadoop_conf.py b/yarn_api_client/hadoop_conf.py
@@ -7,7 +7,7 @@
 
 log = get_logger(__name__)
 
-CONF_DIR = os.getenv('HADOOP_CONF_DIR', '/etc/hadoop/conf')
+CONF_DIR = os.getenv('YARN_CONF_DIR', os.getenv('HADOOP_CONF_DIR', '/etc/hadoop/conf'))
 
 
 def _get_rm_ids(hadoop_conf_path):
